bguisard commited on
Commit
1001ad6
1 Parent(s): edf436d

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  jax[cuda]
2
  jaxlib
3
  flax
 
1
+ -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
2
  jax[cuda]
3
  jaxlib
4
  flax