O que é o Triton (OpenAI) e onde é mais conveniente do que CUDA C++?

Triton é uma linguagem de programação e compilador da OpenAI, projetada para escrever núcleos GPU de alto desempenho, especialmente para tarefas de aprendizagem automática. Permite aos desenvolvedores escrever código semelhante ao Python na sintaxe, mas com uma geração eficiente de código CUDA.

Vantagens do Triton em comparação com CUDA C++:

• Abstração mais elevada e facilidade de escrita de código, o que acelera o desenvolvimento.
• Otimizado para tarefas de ML, como operações matriciais e convoluções.
• Permite experimentar facilmente com núcleos sem mergulhar profundamente nos detalhes do CUDA.

Um exemplo de uso do Triton é escrever núcleos GPU personalizados para otimizar operações no PyTorch.

Assim, o Triton é mais conveniente que o CUDA C++ em situações onde o desenvolvimento rápido e a otimização de operações de ML são importantes sem a necessidade de entender profundamente o CUDA de baixo nível.