torch
torchvision 
einops 
timm
Pillow
hydra
fvcore