absl-py>=2.1.0
attrs>=24.2.0
chex>=0.1.86
dm-haiku>=0.0.12
jax>=0.4.30
jaxlib>=0.4.30
ml_collections>=0.1.1
numpy>=1.26.4
opt-einsum>=3.3.0
optax>=0.2.3
six>=1.16.0
tensorflow>=2.17.0
tfds-nightly>=4.9.3.dev202311230044
toolz>=0.12.1
