carrycooldude commited on
Commit
4067e74
·
1 Parent(s): d58b45f

Add requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +9 -0
requirements.txt ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ jax>=0.3.0
2
+ jaxlib>=0.3.0
3
+ flax>=0.5.0
4
+ optax>=0.1.0
5
+ tensorflow-datasets>=4.0.0
6
+ numpy>=1.19.0
7
+ tqdm>=4.0.0
8
+ matplotlib>=3.0.0
9
+ einops>=0.3.0