absl-py
clu
flax
grain
jax[cuda12]
ml-collections
numpy
optax
orbax
sentencepiece
tensorflow
tensorflow_datasets
