Skip to content

Commit

Permalink
Update jax version in requirements.
Browse files Browse the repository at this point in the history 
PiperOrigin-RevId: 368589236
  • Loading branch information
@hbq1
hbq1 authored and DistraxDev committed Apr 15, 2021
1 parent 55f2afb commit 10639c9
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions requirements/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
absl-py>=0.9.0
chex>=0.0.5
# Change the sign once TFP is fixed
jax<=0.2.11
jax>=0.2.11
jaxlib>=0.1.37
numpy>=1.18.0
tensorflow-probability>=0.12.1
tensorflow-probability>=0.12.2

0 comments on commit 10639c9

Please sign in to comment.