apex
flash-attn
natten
torch
torchvision
transformer-engine
triton
xformers