pybind11
numpy
torch>=1.5.1,<1.6.0
