absl-py>=0.9.0
typing_extensions>=4.2.0
dm-tree>=0.1.5
jax>=0.4.6
jaxlib>=0.1.37
numpy>=1.25.0
toolz>=0.9.0
