jax
astropy
numpyro
matplotlib
