aglib/meliad/requirements.txt

absl-py>=1.0.0
clu>=0.0.7
gin-config>=0.5.0
flax>=0.5.0
jax>=0.3.13
optax>=0.1.2
numpy>=1.22.4
sentencepiece>=0.1.96
seqio>=0.0.7
tensorflow>=2.9.1
tensorflow-datasets>=4.5.2