Limit jax version (#229)

secretflow · Jul 3, 2023 · 381b8dc · 381b8dc
1 parent 160172e
commit 381b8dc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -4,6 +4,6 @@ protobuf>=3.19, <4
 cloudpickle>=2.0.0
 multiprocess>=0.70.12.2
 cachetools>=5.0.0
-jax>=0.3.15  # Last version that supports numpy 1.9
-jaxlib>=0.3.15
+jax>=0.3.15,<=0.4.12  # Last version that supports numpy 1.9
+jaxlib>=0.3.15,<=0.4.12
 termcolor>=2.0.0