pybind11
torch
torchvision