AuleTechnologies/Aule-Attention

High-performance FlashAttention-2 for AMD, Intel, and Apple GPUs. Drop-in replacement for PyTorch SDPA. Triton backend for ROCm (MI300X, RDNA3), Vulkan backend for consumer GPUs. No CUDA required.

语言：Zig · GitHub 星标：160 · Fork：8

话题：ai · cuda · ml · pytorch · rocm

在 GitHub 上查看原仓库