Doc fix: remove outdated advice on flax v0.6.10; it was released two years ago.

PiperOrigin-RevId: 799671841

Author: arnoegw

docs/guides/parallel_training/flax_on_pjit.ipynb

@@ -39,20 +39,6 @@
     "**Note:** This guide uses the `--xla_force_host_platform_device_count=8` flag to emulate multiple devices in a CPU environment in a Google Colab/Jupyter Notebook. You don't need this if you are already using a multi-device TPU environment."
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "metadata": {
-    "tags": [
-     "skip-execution"
-    ]
-   },
-   "outputs": [],
-   "source": [
-    "# Once Flax v0.6.10 is released, there is no need to do this.\n",
-    "# ! pip3 install -qq \"git+https://github.com/google/flax.git@main#egg=flax\""
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 2,

docs/guides/parallel_training/flax_on_pjit.md

@@ -34,13 +34,6 @@ Import some necessary dependencies.
 
 **Note:** This guide uses the `--xla_force_host_platform_device_count=8` flag to emulate multiple devices in a CPU environment in a Google Colab/Jupyter Notebook. You don't need this if you are already using a multi-device TPU environment.
 
-```{code-cell} ipython3
-:tags: [skip-execution]
-
-# Once Flax v0.6.10 is released, there is no need to do this.
-# ! pip3 install -qq "git+https://github.com/google/flax.git@main#egg=flax"
-```
-
 ```{code-cell} ipython3
 import os
 os.environ["XLA_FLAGS"] = '--xla_force_host_platform_device_count=8'