jax
jaxlib
zodiax
numpy
matplotlib
numpyro
tensorflow-probability
astropy
tqdm
optimistix
scipy
astroquery
pyoifits
