flash-attn
flash-attn-3-nv
natten
sageattention
torch
torchvision
transformer-engine
triton
xformers