Spaces:
Runtime error
Runtime error
Commit
•
2b32351
1
Parent(s):
9df39b5
Update requirements.txt
Browse files- requirements.txt +1 -2
requirements.txt
CHANGED
|
@@ -6,8 +6,7 @@ dm-haiku==0.0.4
|
|
| 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==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[cuda]<0.3.0
|
|
|
|
| 10 |
ml-collections==0.1.0
|
| 11 |
numpy==1.19.5
|
| 12 |
pandas==1.3.4
|