absl-py~=2.2
clu==0.0.12
flax~=0.12
jax~=0.9
mlcroissant~=1.0
numpy~=2.3
optax~=0.2
sentencepiece~=0.2
jaxtyping~=0.3
tensorflow-cpu~=2.20
tensorflow-datasets~=4.9
grain~=0.2
orbax-checkpoint[gcp]~=0.11
google-cloud-storage~=3.9
