absl-py==1.4.0
flax==0.6.9
numpy==1.23.5
optax==0.1.5
Pillow==10.2.0
tensorflow==2.12.0
tensorflow-datasets==4.9.2