timm
einops
flash-attn
Pillow
transformers