Spaces:
Runtime error
Runtime error
Commit
·
f2db82f
1
Parent(s):
b8fd344
Update requirements.txt
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
|
@@ -6,8 +6,8 @@ dm-haiku==0.0.4
|
|
| 6 |
dm-tree==0.1.6
|
| 7 |
docker==5.0.0
|
| 8 |
immutabledict==2.0.0
|
| 9 |
-
jax==0.2.14
|
| 10 |
-
jaxlib==0.3.7
|
| 11 |
ml-collections==0.1.0
|
| 12 |
numpy==1.19.5
|
| 13 |
pandas==1.3.4
|
|
|
|
| 6 |
dm-tree==0.1.6
|
| 7 |
docker==5.0.0
|
| 8 |
immutabledict==2.0.0
|
| 9 |
+
jax[cuda111]==0.2.14
|
| 10 |
+
jaxlib==0.3.7
|
| 11 |
ml-collections==0.1.0
|
| 12 |
numpy==1.19.5
|
| 13 |
pandas==1.3.4
|