Estados Unidos
SanMar- External
Waymo
Engenheiro(a) Staff Machine Learning, Infraestrutura
Estados Unidos
Mountain View, California, United States · San Francisco, California, United States
Tradução automática. Consulte o original.
Sobre a oportunidade
A Waymo é uma empresa de tecnologia de direção autônoma com a missão de ser o motorista mais confiável do mundo. Desde o seu início como o Google Self-Driving Car Project em 2009, a Waymo tem se concentrado em construir o Waymo Driver — o motorista mais experiente do mundo™ — para melhorar o acesso à mobilidade e, ao mesmo tempo, salvar milhares de vidas atualmente perdidas em acidentes de trânsito. O Waymo Driver impulsiona o serviço de transporte autônomo da Waymo e também pode ser aplicado a uma variedade de plataformas de veículos e casos de uso de produtos. O Waymo Driver já realizou mais de dez milhões de viagens apenas com passageiros, possibilitado por sua experiência de direção autônoma por mais de 100 milhões de milhas em vias públicas e dezenas de bilhões em simulação em mais de 15 estados dos EUA.
Junte-se à nossa equipe de engenharia de infraestrutura de ML que promove simulações multiagentes ultra-realistas de última geração usando modelos de fundação. Nesta função, você trabalhará na interseção de infraestrutura de ML, modelos de fundação e engenharia de simulação, com foco específico em escrever lógica de negócios e de simulação de alto desempenho em JAX/TensorFlow executada diretamente em TPUs para impulsionar ambientes realistas para Aprendizado por Reforço (RL).
O que você fará
- Projetar, construir e otimizar ambientes de simulação realistas e lógica de negócios executada em TPUs usando JAX e TensorFlow. Implementar e otimizar estratégias de paralelismo de dados e modelos em larga escala para treinar e executar modelos de fundação em hardware TPU.
- Colaborar estreitamente com as equipes de modelagem para integrar modelos de fundação em pipelines de simulação.
- Conduzir arquiteturas técnicas e designs de sistemas, desde a engenharia de dados até a execução da simulação, para atingir os objetivos de negócios e de desempenho.
- Analisar sistemas, identificar gargalos de desempenho em aceleradores de ML e otimizar a velocidade de execução de ponta a ponta.
- Traduzir metas de produto e de negócios em requisitos técnicos concretos e entregáveis de sistema.
Qualificações mínimas
- 6+ anos de experiência profissional em engenharia de software, com pelo menos 4 anos focados em infraestrutura de machine learning (escalonamento, treinamento, otimização e implantação de sistemas de ML em larga escala).
- Experiência direta em programação de ML em hardware TPU e GPU usando frameworks como JAX, PyTorch ou TensorFlow.
- Experiência prática comprovada no escalonamento de grandes modelos usando paralelismo de modelo, paralelismo de dados ou técnicas de treinamento distribuído.
- Sólida compreensão de modelos de ML de última geração (por exemplo, transformers autorregressivos) e proficiência prática com ferramentas de perfil de aceleradores de ML para diagnosticar gargalos.
- Capacidade demonstrada de liderar de forma independente iniciativas técnicas ambíguas de ponta a ponta e construir bibliotecas, pipelines e ferramentas de desenvolvedor robustas.
- Fortes habilidades de comunicação verbal e escrita para colaborar efetivamente entre equipes distribuídas e multifuncionais.
Qualificações preferenciais
- Experiência prática em Aprendizado por Reforço (RL), transferência Sim2Real ou Robótica.
- Experiência com frameworks de ML distribuído e aceleradores como GPU/TPU.
- Familiaridade com sistemas de direção autônoma, simulações multiagentes ou modelagem de mundo realista.
A faixa salarial base esperada para esta posição em tempo integral em locais nos EUA está listada abaixo. O pagamento inicial real será baseado em fatores relacionados ao trabalho, incluindo o local de trabalho exato, experiência, treinamento e educação relevantes e nível de habilidade. Seu recrutador pode compartilhar mais sobre a faixa salarial específica para o local da função ou, se a função puder ser realizada remotamente, a faixa salarial específica para o seu local preferido, durante o processo de contratação.
Os funcionários da Waymo também são elegíveis para participar do programa de bônus anual discricionário da Waymo, do plano de incentivo de ações e do generoso programa de benefícios da empresa, sujeito aos requisitos de elegibilidade.
Faixa salarial
$251,000—$310,000 USD
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