Skip to content

Commit 10639c9

Browse files
hbq1DistraxDev
hbq1
authored and
DistraxDev
committed
Update jax version in requirements.
PiperOrigin-RevId: 368589236
1 parent 55f2afb commit 10639c9

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

‎requirements/requirements.txt‎

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
absl-py>=0.9.0
22
chex>=0.0.5
3-
# Change the sign once TFP is fixed
4-
jax<=0.2.11
3+
jax>=0.2.11
54
jaxlib>=0.1.37
65
numpy>=1.18.0
7-
tensorflow-probability>=0.12.1
6+
tensorflow-probability>=0.12.2

0 commit comments

Comments
 (0)