| 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 |