dm-haiku>=0.0.4
flax>=0.3.4
optax>=0.0.9
scikit-learn>=0.24.1
tensorflow-datasets>=4.4.0
