jupyter
numpy
matplotlib
torch
torchvision