torch>1.0.0
numpy

[dev]
