jax
numpyro
matplotlib
numpy
