Estados Unidos
Roku
NVIDIA
Gerente Sênior, Engenharia de Software - JAX
Estados Unidos
Santa Clara, California, United States
Tradução automática. Consulte o original.
Sobre a oportunidade
O JAX é o framework de escolha para as pesquisas de IA mais ambiciosas, e os datacenters onde ele opera estão evoluindo rapidamente. Os supercomputadores da NVIDIA estão se tornando mais heterogêneos, combinando GPUs, CPUs e LPUs, e o JAX precisa evoluir para ter um bom desempenho em todos eles. A NVIDIA busca um Gerente de Engenharia Sênior para definir e conduzir a estratégia de JAX da NVIDIA, coordenando múltiplas equipes e fluxos de trabalho para garantir que o JAX entregue o máximo desempenho em toda a plataforma, desde cargas de trabalho em um único acelerador até centenas de milhares de dispositivos nos maiores supercomputadores já construídos.
Uma parte fundamental desta função é garantir que o JAX esteja pronto desde o primeiro dia para arquiteturas de próxima geração, incluindo Rubin Ultra, LPUs e as CPUs que as impulsionam. Fazer isso bem exige acompanhar para onde a IA está indo. Isso significa apoiar necessidades emergentes em treinamento, pós-treinamento, inferência e robótica, conectando novas capacidades de hardware com as tendências de IA. Venha se juntar a nós para construir a equipe e a tecnologia por trás dos próximos avanços em IA.
O que você fará
- Conduzir a estratégia de contribuição de engenharia em todo o ecossistema JAX — incluindo o núcleo do JAX, XLA, Shardy, MPMD e bases de código de nível de aplicação como MaxText — priorizando onde as contribuições da NVIDIA criam o máximo impacto em desempenho, adoção e influência no ecossistema.
- Promover o trabalho em equipe entre as fronteiras organizacionais para moldar a plataforma de IA da NVIDIA em todas as equipes de compiladores (XLA, TileIR, CUDA), runtime, rede e hardware.
- Construir parcerias profundas com projetos de código aberto essenciais, especialmente a equipe JAX do Google — apoiando nossos parceiros enquanto representamos os interesses da NVIDIA na definição da direção do projeto.
- Projetar processos e definir critérios de sucesso claros para manter as equipes alinhadas aos resultados.
- Criar planos de pessoal, revisar a capacidade da equipe e tomar decisões estratégicas de contratação para se antecipar às necessidades atuais e futuras.
- Liderar, desenvolver e orientar uma organização de engenharia de alto desempenho — desenvolvendo líderes técnicos e criando espaço para sua equipe experimentar, prototipar e construir sistemas inovadores que melhorem produtos e processos.
O que precisamos ver
- Mestrado ou Doutorado em Ciência da Computação, Engenharia da Computação ou área relacionada (ou experiência equivalente).
- 10+ anos de experiência geral em engenharia de software, com 4+ anos em gestão de engenharia ou liderança técnica.
- Sólida base técnica em software de sistema, compiladores, computação de alto desempenho ou frameworks de IA.
- Experiência comprovada na liderança de estratégias e projetos de nível de domínio que abrangem fronteiras organizacionais, construindo consenso, influenciando sem autoridade direta e adaptando a comunicação ao público, desde engenheiros até executivos.
- Histórico de responsabilidade por resultados em múltiplos projetos e equipes, equilibrando concessões e impulsionando o progresso com informações incompletas.
- Experiência na identificação de lacunas no ciclo de desenvolvimento e na definição de fluxos de trabalho entre equipes e organizações para melhorar a eficiência.
- Experiência na liderança de equipes que contribuem para projetos de grande escala com colaboradores externos.
Formas de se destacar
- Experiência prática com JAX e seus sistemas como XLA, HLO, PJRT, Shardy ou o sistema de rastreamento e transformação do JAX.
- Experiência na liderança de equipes trabalhando em infraestrutura de compilador ou runtime para frameworks de deep learning (JAX, PyTorch, TensorFlow ou equivalente).
- Familiaridade com os componentes da plataforma de IA da NVIDIA: programação de GPU, CUDA, cuDNN, NCCL, TensorRT, Transformer Engine e alvos de computação heterogêneos (GPU, CPU, LPU).
- Experiência na condução de esforços de otimização de desempenho para treinamento de IA distribuído em larga escala, pós-treinamento (RLHF, alinhamento), inferência ou cargas de trabalho de robótica (multi-nó, multi-GPU).
- Experiência na condução de implementação de framework ou software em novas arquiteturas de hardware, coordenando equipes de silício, driver, compilador e framework.
Seu salário base será determinado com base em sua localização, experiência e na remuneração de funcionários em cargos semelhantes. A faixa salarial base é de 272,000 USD a 431,250 USD.
Você também será elegível para participação acionária e benefícios nvidia.com/en-us/benefits/.
As candidaturas para esta vaga serão aceitas pelo menos até 7 de agosto de 2026.
Este anúncio refere-se a uma vaga existente.
A NVIDIA utiliza ferramentas de IA em seus processos de recrutamento.
A NVIDIA está comprometida em promover um ambiente de trabalho inclusivo e orgulha-se de ser um empregador que oferece oportunidades iguais. Como valorizamos muito a diversidade em nossos funcionários atuais e futuros, não discriminamos (inclusive em nossas práticas de contratação e promoção) com base em raça, religião, cor, origem nacional, gênero, expressão de gênero, orientação sexual, idade, estado civil, status de veterano, status de deficiência ou qualquer outra característica protegida por lei.
Mais oportunidades com um só perfil
Quer aparecer para várias vagas sem repetir todo o processo? Crie seu perfil e seja encontrado pelas empresas que usam a Nort.
Criar meu perfilContinue explorando
Redwood Materials
CoreWeave
