Skip to content

Commit f3099b4

Browse files
chaoming0625chaoming0625
committed
Restrict JAX and JAXlib versions to below 0.6.0 in requirements files
1 parent 86c7708 commit f3099b4

3 files changed

Lines changed: 5 additions & 5 deletions

File tree

‎requirements-dev.txt‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
numpy
2-
jax
3-
jaxlib
2+
jax<0.6.0
3+
jaxlib<0.6.0
44
absl-py<=2.1.0
55
brainstate<=0.1.0.post20241210
66
braintaichi<=0.0.4

‎requirements-doc.txt‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
tqdm
2-
jax
3-
jaxlib
2+
jax<0.6.0
3+
jaxlib<0.6.0
44
matplotlib
55
numpy
66
scipy

‎requirements.txt‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
numpy
2-
jax
2+
jax<0.6.0
33
tqdm

0 commit comments

Comments
 (0)