absl-py
jax
jaxlib
numpy
packaging
sentencepiece
tensorflow-text
tfds-nightly

[cache-tasks]
apache-beam

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

[test]
pytest
