torch
Pillow
transformers
timm
einops