simonduerr commited on
Commit
df23d91
1 Parent(s): 7e2da24

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -6,8 +6,8 @@ dm-haiku==0.0.5
6
  dm-tree==0.1.6
7
  docker==5.0.0
8
  immutabledict==2.0.0
9
- jax[cuda]
10
- jaxlib==0.3.7
11
  ml-collections==0.1.0
12
  numpy
13
  pandas==1.3.4
 
6
  dm-tree==0.1.6
7
  docker==5.0.0
8
  immutabledict==2.0.0
9
+ https://storage.googleapis.com/jax-releases/jax/jax-0.3.7.tar.gz
10
+ https://storage.googleapis.com/jax-releases/cuda11/jaxlib-0.3.7+cuda11.cudnn805-cp38-none-manylinux2014_x86_64.whl
11
  ml-collections==0.1.0
12
  numpy
13
  pandas==1.3.4