jax>=0.4.14
numpyro>=0.12.0
