Skip to content

Commit e767e56

Browse files
hbq1DistraxDev
hbq1
authored and
DistraxDev
committed
Update JAX version.
PiperOrigin-RevId: 373741558
1 parent d3706ce commit e767e56

1 file changed

Lines changed: 4 additions & 5 deletions

File tree

‎requirements/requirements.txt‎

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

0 commit comments

Comments
 (0)