tensorflow>=2.5.0
docker
nltk
numpy
pandas
torch>=1.8.0
torch-xla
jax
flax
optax
wandb
tensorflow-hub
tqdm
PyYAML
spacy
transformers
scipy
matplotlib
seaborn

[dev]
pytest
pytest-cov
sphinx
sphinx-rtd-theme
myst-parser
black
flake8

[gpu]
tensorflow-gpu>=2.5.0
torch>=1.8.0
