absl-py==1.0.0
clu==0.0.6
flax==0.3.6
ml-collections==0.1.0
numpy==1.22.0
optax==0.1.0
tensorflow==2.11.1
tensorflow-datasets==4.4.0
tensorflow-text==2.7.0
