torch
torchvision
joblib
scikit-learn