absl-py>=0.9.0
chex>=0.0.7
jax<=0.2.21  # remove once TFP is compatible with JAX>=0.2.22
jaxlib>=0.1.67
numpy>=1.18.0
tensorflow-probability>=0.14.1
