File size: 285 Bytes
e8c4ed3
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
numpy>=1.16.4
jax>=0.2.6
jaxlib>=0.1.57
flax>=0.2.2
opencv-python>=4.4.0
Pillow>=7.2.0
pyyaml>=5.3.1
tensorboard>=2.4.0
tensorflow>=2.3.1
tensorflow-hub>=0.11.0
transformers==4.8.2
wandb==0.10.33
tqdm==4.61.2
# pip install git+https://github.com/deepmind/jmp  # mixed precision for JAX