mitorch-models
pytorch_lightning
torch
torchvision
sklearn
