Spaces:
Running
on
T4
Running
on
T4
Commit
•
df23d91
1
Parent(s):
7e2da24
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
|