torch
triton
transformers
flash-attn