future
numpy
torch>=1.0.1
param
