jax<0.5.0,>=0.4.24
colorama
diffrax>=0.4.1
h5py<4.0,>=3.0.0
interpax>=0.3.3
matplotlib<4.0.0,>=3.5.0
mpmath<2.0,>=1.0.0
netcdf4<2.0,>=1.5.4
numpy>=1.20.0
nvgpu
orthax
plotly<6.0,>=5.16
psutil
pylatexenc<3.0,>=2.0
scikit-image
scipy>=1.7.0
termcolor
