pybind11
numpy
torch>=1.6.0,<1.7.0
