torch
torchvision
numpy