pybind11
numpy
torch>=1.1.0
