engenheiro de kernel gpu - cuda, triton e performance de aceleradores

Sobre a vaga

Buscamos uma pessoa engenheira de kernel GPU para atuar em projetos de alta performance, otimizando kernels para aceleradores como GPUs NVIDIA, AWS Trainium e TPUs. Você vai trabalhar com CUDA, Triton e outras ferramentas de ponta, em um time que valoriza autonomia e impacto técnico.

Responsabilidades

• Desenvolver e otimizar kernels GPU para modelos de machine learning.
• Analisar gargalos de performance usando ferramentas como Nsight e NCU.
• Colaborar com o time para integrar otimizações em pipelines de treinamento e inferência.
• Explorar novas tecnologias de aceleração, incluindo AWS Neuron e XLA.

Requisitos

• Experiência sólida com CUDA e programação de GPUs.
• Conhecimento prático de Triton ou outras linguagens de kernel.
• Familiaridade com frameworks como JAX ou PyTorch.
• Entendimento de arquiteturas de aceleradores (NVIDIA, TPU, Trainium).
• Experiência com profiling e otimização de performance (Nsight, NCU).

Diferenciais

• Conhecimento em MLIR, XLA ou compiladores.
• Experiência com AWS Neuron ou Pallas.
• Contribuições em projetos open source de ML.

Benefícios

• Remuneração competitiva.
• Horário flexível e trabalho remoto.
• Ambiente colaborativo e focado em aprendizado contínuo.