accelerate
timm
einops
torch
torchvision
Pillow
transformers
autoawq
--no-build-isolation flash_attn