absl-py==1.0.0
flax==0.3.6
numpy==1.22.0
optax==0.1.0
Pillow==9.3.0
tensorflow==2.9.3
tensorflow-datasets==4.4.0
