torch
torchvision
scikit-learn