torch
torchvision
joblib
sklearn