pybind11
torch
six
regex
numpy