absl-py
babel
editdistance
gin-config
mesh-tensorflow[transformer]>=0.1.13
nltk
numpy
pandas
rouge-score
sacrebleu
scikit-learn
scipy
sentencepiece
seqio-nightly
six>=1.14
tensorflow-text
tfds-nightly
torch
transformers>=2.7.0

[cache-tasks]
apache-beam

[gcp]
gevent
google-api-python-client
google-compute-engine
google-cloud-storage
oauth2client

[test]
pytest
