Estados Unidos
Nominal
Waymo
Engenheiro 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
Waymo é uma empresa de tecnologia de direção autônoma com a missão de ser o motorista mais confiável do mundo. Desde seu início como o Projeto de Carro Autônomo do Google em 2009, a Waymo tem se concentrado em construir o Waymo Driver — O Motorista Mais Experiente do Mundo™ — para melhorar o acesso à mobilidade, ao mesmo tempo em que salva milhares de vidas perdidas em acidentes de trânsito. O Waymo Driver impulsiona o serviço de ride-hail totalmente 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 em dirigir autonomamente 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 para avançar simulações multiagentes ultrarrealistas de ponta usando modelos de fundação. Nesta função, você trabalhará na intersecção de infraestrutura de ML, modelos de fundação e engenharia de simulação, com foco específico na escrita de lógica de negócios e simulação de alto desempenho em JAX/TensorFlow executando diretamente em TPUs para potencializar 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 executando em TPUs usando JAX e TensorFlow. Implementar e otimizar estratégias de paralelismo de modelos e dados em larga escala para treinamento e execução de modelos de fundação em hardware TPU.
- Colaborar estreitamente com equipes de modelagem para integrar modelos de fundação em pipelines de simulação.
- Liderar arquiteturas técnicas e projetos de sistemas desde a engenharia de dados até a execução da simulação para atender aos objetivos de negócios e 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 negócios em requisitos técnicos concretos e entregas de sistemas.
Qualificações Mínimas
- Mais de 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 em escalonamento de modelos grandes usando paralelismo de modelo, paralelismo de dados ou técnicas de treinamento distribuído.
- Forte compreensão de modelos de ML de ponta (por exemplo, transformers autorregressivos) e proficiência prática com ferramentas de profiling de aceleradores de ML para diagnosticar gargalos.
- Capacidade comprovada de liderar de forma independente iniciativas técnicas ambíguas de ponta a ponta e construir bibliotecas robustas, pipelines e ferramentas para desenvolvedores.
- Fortes habilidades de comunicação verbal e escrita para colaborar efetivamente em 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ídos 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 salário inicial real será baseado em fatores relacionados ao trabalho, incluindo 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 vaga ou, se a função puder ser realizada remotamente, a faixa salarial específica para 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, plano de incentivo acionário e 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