numpy
torch>=1.0.1
param
