jax>=0.4.13
astropy
numpy
matplotlib
diffrax
chainconsumer
emcee
h5py
pandas
scipy
seaborn
tqdm
jax-cosmo
tinygp
pytest