absl-py
clu
editdistance
jax
jaxlib
numpy
packaging
pyglove
sentencepiece>=0.2
tensorflow-text
tfds-nightly==4.9.2.dev202308090034

[cache-tasks]
apache-beam

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

[test]
pytest
