Spaces:
Running
on
T4
Running
on
T4
simonduerr
commited on
Commit
•
6078146
1
Parent(s):
0f3c708
Update requirements.txt
Browse files- 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
|
10 |
-
jaxlib
|
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
|