Bump JAX requirements.

PiperOrigin-RevId: 415386111
google-deepmind · Dec 10, 2021 · 3f15527 · 3f15527
1 parent f874857
commit 3f15527
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements-jax.txt b/requirements-jax.txt
@@ -1,2 +1,2 @@
-jax==0.2.22
-jaxlib==0.1.73
+jax>=0.2.22
+jaxlib>=0.1.73