absl-py>=0.12.0 numpy>=1.21.5 tensorflow-cpu>=2.12.0 tensorflow-datasets>=4.4.0 matplotlib>=3.5.0 clu>=0.0.3 flax==0.6.9 chex>=0.0.7 optax>=0.1.0 ml-collections>=0.1.0 scikit-image sunds