numpy
torch>=1.4,<1.7