Spaces:
Running
on
T4
Running
on
T4
Commit
·
79b5885
1
Parent(s):
8826366
Update requirements.txt
Browse files- requirements.txt +3 -2
requirements.txt
CHANGED
|
@@ -5,8 +5,8 @@ dm-haiku==0.0.5
|
|
| 5 |
dm-tree==0.1.6
|
| 6 |
docker==5.0.0
|
| 7 |
immutabledict==2.0.0
|
| 8 |
-
jax[cuda]
|
| 9 |
-
jaxlib
|
| 10 |
ml-collections==0.1.0
|
| 11 |
numpy==1.19.5
|
| 12 |
pandas==1.3.4
|
|
@@ -19,4 +19,5 @@ ray
|
|
| 19 |
tqdm
|
| 20 |
gradio==3.0.11
|
| 21 |
protobuf<4
|
|
|
|
| 22 |
-f https://storage.googleapis.com/jax-releases/jax_releases.html
|
|
|
|
| 5 |
dm-tree==0.1.6
|
| 6 |
docker==5.0.0
|
| 7 |
immutabledict==2.0.0
|
| 8 |
+
jax[cuda]==0.3.8
|
| 9 |
+
jaxlib==0.3.7
|
| 10 |
ml-collections==0.1.0
|
| 11 |
numpy==1.19.5
|
| 12 |
pandas==1.3.4
|
|
|
|
| 19 |
tqdm
|
| 20 |
gradio==3.0.11
|
| 21 |
protobuf<4
|
| 22 |
+
git+https://github.com/Acellera/moleculekit.git
|
| 23 |
-f https://storage.googleapis.com/jax-releases/jax_releases.html
|