Software Development Engineer - Triton Kernel Development

Sobre a vaga

A AMD busca um Engenheiro de Software para atuar no desenvolvimento e otimização de Triton Kernels em São Paulo, Brasil. Esta é uma oportunidade para trabalhar diretamente com tecnologias de ponta em computação de alto desempenho e inteligência artificial.

Responsabilidades

• Desenvolver, otimizar e validar Triton kernels para aceleração de cargas de trabalho de IA e aprendizado de máquina.
• Trabalhar na integração e melhoria de desempenho em ecossistemas como PyTorch e TensorFlow sobre a plataforma ROCm.
• Colaborar com equipes globais de engenharia para solucionar problemas complexos de compilação e execução de código.

Requisitos

• Experiência sólida em programação com C++ e Python.
• Familiaridade com frameworks de aprendizado de máquina como PyTorch ou TensorFlow.
• Conhecimento em arquiteturas de GPU, programação paralela (CUDA, HIP) ou otimização de baixo nível (Assembly, Compute Kernel).
• Boa capacidade de comunicação e trabalho em equipe em ambiente internacional.

Diferenciais

• Experiência prévia com o compilador LLVM, Triton ou bibliotecas como CUTLASS.
• Conhecimento em computação de alto desempenho (HPC) e otimização de kernels.