updating jax version

PAIR-code · Nov 8, 2023 · 18e71ce · 18e71ce
1 parent b421a61
commit 18e71ce
Showing 1 changed file with 2 additions and 2 deletions.
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
+jax==0.4.1
+jaxlib==0.4.1
 flax==0.5.3
 # LINT.ThenChange(./pyproject.toml)