numpy
h5parm>=1.0.5
scipy
astropy>=6
matplotlib
cmocean
tqdm
jax
jaxlib
tensorflow_probability
