Triton OpenAI Triton
Mer info
- Skapad av
- Philippe Tillet (Harvard / OpenAI)
- År
- Typ
- Python-likt språk för GPU-kernels
- Licens
- MIT
- Källkod
- github.com/triton-lang/triton
Python-DSL för att skriva CUDA-kernels — Tillet vid OpenAI. Driver FlashAttention, PyTorch 2:s torch.compile, och de flesta moderna LLM-träningar.
Pitch: skriv kernels på Python-nivå (utan att hantera shared memory, banks, warps manuellt), Triton-compiler genererar konkurrerande PTX. Tri Dao (FlashAttention) skrev både Cutlass- och Triton-versioner — Triton vann i läsbarhet. AMD ROCm-stöd kom 2024. Inte att förväxla med Nvidia Triton Inference Server (helt annan grej).