matplotlib
numpy
# torch
# torchvision