Skip to content

Commit

Permalink
small
Browse files Browse the repository at this point in the history
  • Loading branch information
@amacrutherford
amacrutherford committed Mar 27, 2024
1 parent 1313ff2 commit e0d8260
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion 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
# jaxlib==0.4.17 # commenting these out as it causes the dockerfile to reinstall jax as only the cpu version >:|
flax==0.7.4
chex==0.1.84
optax==0.1.7
Expand Down

0 comments on commit e0d8260

Please sign in to comment.