torch
torch
torchvision
matplotlib
