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