Cython
torch
torchvision