pybind11>=2.6.0
torch
