[rlax] Update jax and numpy requirements for RLax.

PiperOrigin-RevId: 500667041
google-deepmind · Jan 9, 2023 · a6137d7 · a6137d7
1 parent 44ef3f0
commit a6137d7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements/requirements.txt b/requirements/requirements.txt
@@ -2,6 +2,6 @@ absl-py>=0.9.0
 chex>=0.0.8
 distrax>=0.0.2
 dm_env
-jax>=0.1.55
+jax>=0.3.0
 jaxlib>=0.1.37
-numpy>=1.18.0,<1.23
+numpy>=1.18.0