simonduerr commited on
Commit
8826366
·
1 Parent(s): 70eab4f

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -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]==0.3.8
9
- jaxlib==0.3.7
10
  ml-collections==0.1.0
11
  numpy==1.19.5
12
  pandas==1.3.4
 
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