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