jax==0.2.13
flax==0.2.2
ml-collections>=0.1.0
tensorboard>=2.3.0
tensorflow==2.7.0
tensorflow-datasets>=4.0.1
tensorflow_text==2.7.0
gin
gin-config>=0.1.3
attrs