numpy>=1.16.5
scipy
torch
