torch
math
torchvision
matplotlib
numpy
scikit-learn