accelerate
timm
einops
torch
torchvision
Pillow
transformers
autoawq
flash_attn