Skip to content

Commit

Permalink
removing jax requiremetn
Browse files Browse the repository at this point in the history
  • Loading branch information
@amacrutherford
amacrutherford committed Mar 27, 2024
1 parent e73e470 commit 1313ff2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions requirements/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# requirements are alligned with nvcr.io/nvidia/jax:23.10-py3 image
jax==0.4.17
jaxlib==0.4.17
# jax==0.4.17
# jaxlib==0.4.17
flax==0.7.4
chex==0.1.84
optax==0.1.7
Expand Down

0 comments on commit 1313ff2

Please sign in to comment.