RIFT>=0.0.15.11
arviz>=0.20.0
chex>=0.1.87
corner>=2.2.3
equinox>=0.11.3
flowMC==0.4.5
glasbey>=0.3.0
h5py>=3.12.1
jaxtyping>=0.2.36
jenkspy==0.4.1
loguru>=0.7.0
matplotlib>=3.9.0
mplcursors>=0.6
nbconvert>=7.17.1
numpy<3
numpyro>=0.19.0
optax<0.2.7
pandas>=2.2.0
papermill>=2.7.0
plotly>=6.7.0
pydantic>=2.12.0
quadax>=0.2.5
rich>=14.0.0
scipy>=1.12
seaborn>=0.13.0
tqdm>=4.60.0

[cpu]
jax>=0.7.0

[cuda12]
jax[cuda12]>=0.7.0

[cuda13]
jax[cuda13]>=0.7.0

[tpu]
jax[tpu]>=0.7.0
