absl-py>=2.3.1
chex>=0.1.90
jax>=0.7.0
jaxlib>=0.7.0
numpy>=1.24.1
setuptools;python_version>="3.12"
tensorflow-probability>=0.25.0
