scikit-learn
torch
numpy
torchvision
