updating jax version

jameswex · jameswex · commit 18e71ce246a0 · 2023-11-08T10:19:54.000-05:00
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)
