transformers
einops
timm 
pillow