simonduerr commited on
Commit
abce251
1 Parent(s): 4393748

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -6,7 +6,7 @@ dm-haiku==0.0.5
6
  dm-tree==0.1.6
7
  docker==5.0.0
8
  immutabledict==2.0.0
9
- jax[cuda]==0.3.8
10
  jaxlib==0.3.7
11
  ml-collections==0.1.0
12
  numpy==1.19.5
 
6
  dm-tree==0.1.6
7
  docker==5.0.0
8
  immutabledict==2.0.0
9
+ jax[cuda]
10
  jaxlib==0.3.7
11
  ml-collections==0.1.0
12
  numpy==1.19.5