Estados Unidos
CONCEPT SOLUTIONS LLC
NVIDIA
Engenheiro de Software Sênior — cuEquivariance
Estados Unidos
Santa Clara, California, Estados Unidos
PythonC++
Tradução automática. Consulte o original.
Sobre a oportunidade
NVIDIA está transformando gráficos de computador, jogos de PC e computação acelerada há mais de 25 anos. É um legado único de inovação impulsionado por ótima tecnologia — e pessoas incríveis. Hoje, estamos aproveitando o potencial ilimitado da IA para definir a próxima era da computação. Uma era em que nossa GPU atua como o cérebro de computadores, robôs e carros autônomos que podem entender o mundo. Fazer o que nunca foi feito antes exige visão, inovação e os melhores talentos do mundo. Como um NVIDIAN, você será imerso em um ambiente diversificado e de apoio, onde todos são inspirados a fazer o seu melhor trabalho. Junte-se ao nosso grupo e descubra como você pode causar um impacto duradouro no mundo.
NVIDIA BioNeMo está construindo a base computacional para a próxima geração de descobertas biológicas. Estamos procurando um Engenheiro de Software Sênior para se juntar à equipe cuEquivariance — uma biblioteca NVIDIA que acelera redes neurais geométricas em GPUs NVIDIA, permitindo que pesquisadores em biologia molecular, ciência de materiais e física treinem e implementem modelos equivariantes em escala. Esta equipe constrói e envia os kernels de GPU de produção e as interfaces de software que impulsionam deep learning equivariantes em todo o campo científico. O trabalho abrange engenharia de kernels CUDA, desenvolvimento de bibliotecas Python envolvendo PyTorch e JAX, e colaboração direta com equipes de pesquisa e desenvolvedores de frameworks externos. Se você quer trabalhar onde a computação de GPU encontra deep learning baseada em grafos, esta é a função para você. Seu trabalho será executado em pipelines de produção em toda a comunidade científica.
O que você fará
- Construir, implementar e otimizar kernels CUDA para primitivas de redes neurais equivariantes — produtos tensoriais, polinômios segmentados e operações baseadas em triângulos — visando o desempenho máximo em gerações de GPU NVIDIA.
- Ser responsável pela entrega ponta a ponta de primitivas de ML geométricas aceleradas por GPU: da implementação a software validado e de qualidade de produção do qual frameworks externos dependem.
- Construir e manter as interfaces para PyTorch e JAX que expõem as primitivas cuEquivariance a desenvolvedores de aplicativos e pesquisadores.
- Liderar a infraestrutura CI/CD para compilações de kernels multi-GPU, testes automatizados de correção e rastreamento de regressão de desempenho.
- Colaborar com equipes de Ciência Aplicada e pesquisa para avaliar novas arquiteturas equivariantes e traduzir protótipos em kernels de produção.
- Engajar diretamente com desenvolvedores de frameworks de terceiros e parceiros para alinhar interfaces e garantir que o software entregue se integre perfeitamente aos pipelines de produção.
O que precisamos ver
- 6+ anos de experiência em engenharia de software com forte base em CUDA e programação de GPU.
- Proficiência profunda em C++ e Python; experiência na construção e envio de bibliotecas de produção usadas por desenvolvedores externos.
- Boa base em computação de GPU: hierarquia de memória, execução de warp, ocupação e metodologia de profiling de desempenho.
- Experiência na construção ou contribuição para bibliotecas de software científico de produção, frameworks de ML ou APIs de GPU voltadas para desenvolvedores.
- Familiaridade com conceitos em machine learning geométrica — equivariância, representações de grupo, representações irredutíveis ou produtos tensoriais — suficiente para trabalhar eficientemente no domínio.
- Bacharelado/Mestrado em Ciência da Computação, Física, Matemática Aplicada ou área relacionada, ou experiência equivalente.
Formas de se destacar
- Você contribuiu ou usou profundamente um framework de rede neural importante que respeita a equivariância: e3nn, MACE, NequIP, SE(3)-Transformers ou similar.
- Experiência prática com desenvolvimento de kernels Triton ou outras ferramentas de autoria de kernels de GPU juntamente com CUDA.
- Experiência com design de algoritmos de precisão mista ou ciente de tensor-core para cargas de trabalho científicas ou de ML.
- Doutorado ou experiência equivalente em química computacional, biofísica, física ou ciência da computação com foco em deep learning geométrica ou HPC.
- Contribuições para projetos de ML geométrico ou computação de GPU de código aberto.
Considerada amplamente um dos empregadores mais desejados do mundo da tecnologia, a NVIDIA oferece salários altamente competitivos e um pacote abrangente de benefícios. Ao planejar seu futuro, veja o que podemos oferecer a você e sua família nvidiabenefits.com/ nvidiabenefits.com/
Seu salário base será determinado com base em sua localização, experiência e remuneração de funcionários em posições semelhantes. A faixa salarial base é 184,000 USD - 287,500 USD para o Nível 4, e 224,000 USD - 356,500 USD para o Nível 5.
Você também será elegível para ações e benefícios nvidia.com/en-us/benefits/.
As inscrições para esta vaga serão aceitas pelo menos até 21 de agosto de 2026.
Esta publicação é para uma vaga existente.
NVIDIA utiliza ferramentas de IA em seus processos de recrutamento.
NVIDIA está comprometida em promover um ambiente de trabalho inclusivo e se orgulha de ser um empregador que oferece igualdade de oportunidades. Como valorizamos muito a diversidade em nossos funcionários atuais e futuros, 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, 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 perfil