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