scikit-learn
pytorch
torchvision
tensorboardX
