colorama
h5py>=3.0.0
jax[cpu]<=0.4.14,>=0.3.2
matplotlib!=3.4.3,<=3.6.0,>=3.3.0
mpmath>=1.0.0
netcdf4>=1.5.4
numpy<1.25.0,>=1.20.0
nvgpu
psutil
scipy<1.11.0,>=1.5.0
termcolor
