torch
einops
flash-attn
accelerate
bitsandbytes
