Bump jaxlib version to 0.4.13 and flax version to 0.6.11.

PiperOrigin-RevId: 579944240
PAIR-code · Nov 6, 2023 · f254fa8 · f254fa8
1 parent 688a8cd
commit f254fa8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/lit_nlp/examples/dalle/requirements.txt b/lit_nlp/examples/dalle/requirements.txt
@@ -15,9 +15,9 @@
 
 -r ../../../requirements_core.txt
 
-jax==0.3.25
-jaxlib==0.3.25
-flax==0.6.3
+jax==0.4.13
+jaxlib==0.4.13
+flax==0.6.11
 dalle-mini==0.1.5
 ipywidgets==7.5.0
 orbax==0.0.23

diff --git a/requirements_examples.txt b/requirements_examples.txt
@@ -18,7 +18,7 @@ sentencepiece==0.1.99
 tensorflow-datasets==4.8.0
 torch==2.0.1
 transformers==4.27.1
-jax==0.3.16
-jaxlib==0.3.15
-flax==0.5.3
+jax==0.4.13
+jaxlib==0.4.13
+flax==0.6.11
 # LINT.ThenChange(./pyproject.toml)