tensorflow
torch
matplotlib
numpy
torchvision