tensorflow==2.12.0
scipy==1.9.3
numpy
ml_collections
absl-py