Principal Product Manager, AI Frameworks

Na NVIDIA, encontramos clientes onde eles estão em sua jornada de IA em nossas GPUs - isso significa que construímos melhor em frameworks de classe na OSS e suportamos um ecossistema robusto de outros frameworks da OSS. A equipe de Compiladores PyTorch da NVIDIA constrói e a montante para a pilha que fica entre o hardware PyTorch e NVIDIA — tocha que se estende. compilar, emergente substratos compiladores e a infraestrutura de otimização agente-nativo sendo construída para a próxima era de computação acelerada. Este papel construirá e dirigirá a estratégia do produto em todo o arco. Envolve o envio dos recursos de hardware mais recentes em tocha. compilar hoje. Ele também inclui o desenvolvimento da representação compartilhada canônica através do compilador NVIDIA e ecossistema em tempo de execução. Além disso, foca-se na criação de como os agentes se envolverão no trabalho de desempenho de aprendizagem profunda no futuro. Procuramos alguém que entenda compiladores e possa operar na intersecção de arquitetura de sistemas, engenharia de framework e estratégia de produtos voltados para o cliente trabalhando diretamente com a liderança de engenharia e os clientes externos mais sofisticados da NVIDIA — incluindo laboratórios de modelos de fronteira, equipes de infraestrutura de inferência/treinamento e RL, como vLLM, SGLang, tochatitan, megatron-core e programas de co-design de hardware. Como gerentes de produtos da NVIDIA, somos parceiros com líderes da NVIDIA para definir estratégias claras de produtos e equipes de equipes de marketing para construir planos de go-to-market. A organização de gerenciamento de produtos da NVIDIA é um grupo flexível, forte e impactante com foco em permitir uma aprendizagem profunda em todos os casos de uso de GPU e fornecer ótimos produtos para nossos usuários. Procuramos um indivíduo com uma rara mistura de habilidades de produto, profundidade técnica e paixão para se juntar à nossa equipe. Parece-te familiar? Em caso afirmativo, adoraríamos ouvir de você! O que vais fazer? Possuir a estratégia para a carteira de compiladores PyTorch da NVIDIA, incluindo: • tocha.compile — mantenha e evolua o caminho upstream da NVIDIA PyTorch, ative suporte HW, resolva problemas do cliente em formas dinâmicas, desempenho do kernel e compile sobrecarga • definir o roadmap e ir ao mercado para a camada de representação compartilhada da NVIDIA entre frameworks, compiladores, bibliotecas de kernel e runtimes • Fluxos de trabalho do compilador agente-nativo — modelar a visão do produto para como os sistemas agentes participarão em tarefas de otimização • Recursos de otimização inter-kernel — construir requisitos de produto para capacidades como megakernels e garantir que eles enviam com valor do mundo real demonstrado Estratégia do produto líder em todo o roteiro: • Definir prioridades agora/próximo/mais tarde em estreita parceria com líderes de engenharia • Traduzir sinais ecossistêmicos (vLLM RFCs, SGLang CUDA Graph patterns, TorchTitan GraphTrainer plans, Meta's upstream priorities) em decisões priorizadas do produto Interaja diretamente com clientes e parceiros: • Representar a NVIDIA em fóruns de contribuição e ecossistema da PyTorch; Defina métricas de sucesso e critérios de lançamento: • Estabelecer portas de desempenho e marcos de adoção • Definir barra para o que "provado" significa para novas otimizações antes de se comprometer com o roteiro O que precisamos de ver: • 15 + anos em gestão técnica de produtos, com 5 anos de posse de um compilador, tempo de execução, ou de sistemas de baixo nível produto em escala • BS ou mestrado em Ciência da Computação, Engenharia Elétrica, um campo técnico relacionado, ou experiência equivalente. • Experiência com os primeiros produtos da OSS e estratégia de contribuição a montante • Track registro de transporte e condução adoção para a eficiência do desenvolvedor e produtos de infraestrutura orientada para o desempenho • Entende como os frameworks de inferência usam a tecnologia do compilador – onde adotam tocha. compilar, onde eles vão em torno dele, e por que • Entende como novos recursos de hardware criam novos requisitos de compilador • Pode escrever documentos de estratégia claros e defensáveis e sabe como explorar um lançamento de acesso antecipado: • Forte instinto para onde concentrar investimento vs. disseminá-lo Maneiras de se destacar da multidão: • Conhecimento profundo da pilha de compiladores PyTorch e como influenciar a estratégia neste ecossistema • Você trabalhou em como os agentes participam de fluxos de trabalho de otimização de nível de sistemas • Familiaridade com a infraestrutura do compilador baseada em MLIR e como ele mapeia para os primitivos hardware NVIDIA • Pode raciocinar sobre tradeoffs de otimização inter-kernel para definir a barra certa para "provado" • Perfis de desempenho de leitura confortável do kernel e depurar como tocha. compilar pode ajudar qualquer modelo LI-Hybrid Seu salário base será determinado com base na sua localização, experiência e o pagamento de funcionários em posições semelhantes. A faixa salarial base é de 240.000 USD - 379.500 USD. Você também será elegível para equidade e benefícios. Os pedidos para este trabalho serão aceitos pelo menos até 27 de julho de 2026. Esta postagem é para uma vaga existente. NVIDIA usa ferramentas de IA em seus processos de recrutamento. A NVIDIA está empenhada em promover um ambiente de trabalho inclusivo e orgulhar-se de ser um empregador de igualdade de oportunidades. Como valorizamos a diversidade em nossos atuais e futuros funcionários, não discriminamos (incluindo 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, condição de deficiência ou qualquer outra característica protegida por lei.