h5py
scipy
numpy
torch>=1.8.1
