Use tf-nightly for Haiku testing.

PiperOrigin-RevId: 607673638
google-deepmind · Feb 16, 2024 · 2af2f19 · 2af2f19
1 parent 134763d
commit 2af2f19
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.16
-jaxlib>=0.4.16
+jax>=0.4.24
+jaxlib>=0.4.24
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
-tensorflow>=2.9.1
+tf-nightly>=2.16.0
 optax>=0.0.1