pybind11
numpy
torch>=1.5.0
