pybind11
numpy
torch<1.6.0,>=1.5.1
