absl-py
clu
editdistance
jax
jaxlib
numpy
packaging
pyglove
sentencepiece
tensorflow-text
tfds-nightly
protobuf<=3.20.3

[cache-tasks]
apache-beam

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

[test]
pytest
