pybind11
torch
torchvision
six
regex
numpy