jax>=0.3.1
jaxlib>=0.3.1
numpyro>=0.9.0
blackjax==0.7.0