tensorflow>=2.5.0
docker
nltk
numpy
pandas
torch>=1.8.0
torch-xla
jax
wandb
