absl-py==1.0.0
clu==0.0.6
flax==0.3.6
numpy==1.22.0
optax==0.1.0
