torch 
torchvision 
einops 
timm