simonduerr commited on
Commit
49b5296
1 Parent(s): 3f1a884

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -7,11 +7,12 @@ dm-tree==0.1.6
7
  docker==5.0.0
8
  immutabledict==2.0.0
9
  jax[cuda]<0.3.0
10
- https://storage.googleapis.com/jax-releases/cuda111/jaxlib-0.1.72+cuda111-cp37-none-manylinux2010_x86_64.whl
11
  ml-collections==0.1.0
12
  numpy==1.19.5
13
  pandas==1.3.4
14
  scipy==1.7.0
15
  tensorflow-gpu==2.5.0
16
  torch
17
- plotly
 
 
7
  docker==5.0.0
8
  immutabledict==2.0.0
9
  jax[cuda]<0.3.0
10
+ jaxlib==0.1.76
11
  ml-collections==0.1.0
12
  numpy==1.19.5
13
  pandas==1.3.4
14
  scipy==1.7.0
15
  tensorflow-gpu==2.5.0
16
  torch
17
+ plotly
18
+ -f https://storage.googleapis.com/jax-releases/