torch
torchvision
joblib