Bump required JAX and TF versions to latest.

Required to land #779. PiperOrigin-RevId: 633848479
google-deepmind · May 15, 2024 · 51c339e · 51c339e
1 parent 48dcadd
commit 51c339e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/requirements-jax.txt b/requirements-jax.txt
@@ -1,2 +1,2 @@
-jax>=0.4.24
-jaxlib>=0.4.24
+jax>=0.4.28
+jaxlib>=0.4.28
diff --git a/requirements-test.txt b/requirements-test.txt
@@ -5,5 +5,5 @@ mock>=3.0.5
 pytest-xdist
 virtualenv
 dm-tree>=0.1.1
-tf-nightly>=2.16.0
+tensorflow>=2.16.1
 optax>=0.0.1