pybind11
numpy
torch<1.7.0,>=1.6.0
