absl-py==1.0.0
flax==0.3.6
numpy==1.22.0
tensorflow==2.11.1
