numpy
scipy
h5py
numba
torch>=1.7.0
dask
dask-jobqueue
dask-kubernetes
matplotlib
gputil
