tensorflow>=2.5.0
docker
nltk
numpy
pandas
torch>=1.8.0
torch-xla
jax
flax
optax
wandb
tensorflow-hub
tqdm
PyYAML
