jax!=0.7.1,<0.10.0,>=0.6.2
colorama<=0.4.6
diffrax<=0.7.2,>=0.6.0
equinox<=0.13.8,>=0.11.10
h5py<=3.16.0,>=3.0.0
interpax<0.4,>=0.3.3
interpax_fft<=0.0.9,>=0.0.9
lineax
jax-finufft<=1.3.1,>=1.1.0
matplotlib<=3.10.8,>=3.7.3
mpmath<=1.4.1,>=1.0.0
netcdf4!=1.7.4,<=1.7.5,>=1.5.4
numpy<=2.5,>=1.20.0
nvidia-ml-py>=12.535.77
optax<0.3
orthax<0.3,>=0.2.7
plotly<=6.7.0,>=5.16
psutil<=7.2.2
pylatexenc<=2.10,>=2.0
quadax<0.3,>=0.2.2
scikit-image<=0.26.0
scipy<=1.17.1,>=1.7.0
sympy
termcolor<=3.3.0
