numpy
einops
flash_attn
Pillow
gradio
transformers
timm