pybind11
numpy
torch>=1.4.0
