accelerate
torch
flash_attn
transformers
