numpy
Pillow
Requests
flash_attn
transformers
einops