Update configure_flax.md

NbAiLab · Dec 26, 2021 · 6020317 · 6020317
1 parent 4c8c5a2
commit 6020317
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/guides/configure_flax.md b/guides/configure_flax.md
@@ -57,7 +57,7 @@ source flax/bin/activate
 Run this on the VM. Dont worry if the first command returns a few errors.:
 ```bash
 pip install --upgrade clu
-pip install "jax[tpu]>=0.2.16" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
+pip install "jax[tpu]>=0.2.26" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
 export USE_TORCH=False
 ```