absl-py>=0.9.0
typing_extensions>=4.2.0
jax>=0.4.27
jaxlib>=0.4.27
numpy>=1.24.1
toolz>=0.9.0

[:python_version >= "3.12"]
setuptools
