torch>=1.10
pybind11
ninja
