absl-py>=2.3.1
chex>=0.1.90
jax>=0.7.0
jaxlib>=0.7.0
numpy>=1.24.1
tensorflow-probability>=0.25.0

[:python_version >= "3.12"]
setuptools
