Estados Unidos
The MITRE Corporation
Waymo
Staff Engenheiro(a) de ML, Performance e Eficiência de Modelos Generativos
Estados Unidos
Mountain View, California, United States · New York, New York, United States
PythonC++
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 Motorista Waymo—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 Motorista Waymo 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 Motorista Waymo 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.
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 Motorista Waymo—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 Motorista Waymo 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 Motorista Waymo 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.
A Equipe de Simuladores da Waymo constrói simulações de ponta de ambientes realistas para teste, treinamento e validação do Motorista Waymo. Nossa equipe é um grupo diversificado e colaborativo de engenheiros de machine learning (ML), engenheiros de software e engenheiros de pesquisa de ML. Desenvolvemos soluções de simulação líderes na indústria usando algoritmos avançados de ML generativos e reconstrutivos, para modelar o mundo real, abrangendo agentes realistas, estradas, sistemas de tráfego, clima e o conjunto completo de sensores (Câmera, Lidar, Radar).
Para acelerar a fidelidade, escalabilidade, controlabilidade e riqueza de nossas simulações, estamos expandindo as fronteiras da modelagem do mundo em 3D. Aproveitamos tecnologias de ML de ponta treinadas em conjuntos de dados em larga escala para criar mundos virtuais dinâmicos e semanticamente ricos, impactando diretamente o desenvolvimento e a validação do Motorista Waymo.
Nesta função, você se reportará a um Gerente Sênior de Engenharia Staff.
Você irá
- Analisar arquiteturas de modelos e identificar gargalos no desempenho de treinamento e inferência (por exemplo, largura de banda de memória, computação, comunicação).
- Aplicar e desenvolver técnicas como quantização (por exemplo, FP8, INT4), poda, destilação de conhecimento e mecanismos de atenção eficientes.
- Otimizar o código do modelo para aceleradores de hardware específicos (TPUs, GPUs), aproveitando recursos do compilador e bibliotecas de baixo nível (por exemplo, XLA).
- Experimentar diferentes estratégias de particionamento e fragmentação de modelos (por exemplo, paralelismo de dados, tensor, pipeline, paralelismo de especialistas) para melhorar a escalabilidade e a eficiência.
- Projetar e implementar soluções de serviço de baixa latência e alta taxa de transferência para modelos generativos e otimizar pipelines de treinamento para reduzir o tempo de treinamento.
- Construir e manter ferramentas para análise de desempenho, profiling (por exemplo, xprof) e depuração de modelos de ML.
Você tem
- Mestrado ou Doutorado em Ciência da Computação, Machine Learning, Robótica ou área relacionada.
- 5+ anos de experiência com arquiteturas de deep learning (especialmente Transformers, Modelos de Difusão, MoEs), algoritmos e técnicas de otimização.
- Proficiência em JAX, Flax e potencialmente TensorFlow/PyTorch.
- Experiência em usar ferramentas de profiling (por exemplo, XProf, Perfetto, NVIDIA Nsight) para diagnosticar problemas de desempenho em cargas de trabalho de ML.
- Experiência prática com quantização, poda, destilação e outros métodos de compressão de modelos.
- Fortes habilidades de programação em Python e potencialmente C++, com experiência em melhores práticas de desenvolvimento de software.
Preferimos
- Conhecimento de arquiteturas TPU e GPU e como otimizar código para elas.
- Familiaridade com compiladores de ML como XLA e compreensão de como eles traduzem código de alto nível em instruções de hardware eficientes.
- Compreensão de conceitos relacionados ao treinamento e serviço de modelos em vários dispositivos e máquinas.
- Experiência em contribuir para frameworks e bibliotecas que melhoram a velocidade e a escalabilidade do treinamento (por exemplo, JAX, Gemax, XManager).
A faixa salarial base esperada para esta posição em tempo integral em locais dos EUA está listada abaixo. O salário inicial real será baseado em fatores relacionados ao trabalho, incluindo o local exato de trabalho, experiência, treinamento e educação relevantes, e nível de habilidade. Seu recrutador poderá compartilhar mais informações sobre a faixa salarial específica para a localização da vaga ou, se a vaga puder ser realizada remotamente, a faixa salarial específica para sua localização preferida, durante o processo de contratação.
Faixa Salarial $251,000—$310,000 USD
Os funcionários da Waymo também são elegíveis para participar do programa discricionário de bônus anual da Waymo, plano de incentivo acionário e generoso programa de benefícios da Empresa, sujeito aos requisitos de elegibilidade.
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