Update chex & numpy versions in requirements. Closes #234.

PiperOrigin-RevId: 544330735
google-deepmind · Jun 29, 2023 · aa74619 · aa74619
1 parent 1a40423
commit aa74619
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements/requirements.txt b/requirements/requirements.txt
@@ -1,6 +1,6 @@
 absl-py>=0.9.0
-chex>=0.0.7
+chex>=0.1.8
 jax>=0.1.55
 jaxlib>=0.1.67
-numpy>=1.18.0,<1.23
+numpy>=1.23.0
 tensorflow-probability>=0.15.0