torch
torchvision
timm
yacs
joblib
numpy