torch>=1.8.1
numpy
scipy
matplotlib
h5py
