A Zoox desenvolve tecnologia de mobilidade autônoma e uma frota criada especificamente para oferecer mobilidade como serviço. Seu trabalho reúne robótica, inteligência artificial, engenharia automotiva, veículos elétricos e sistemas avançados de sensoriamento para criar um transporte mais seguro, limpo e agradável. A empresa busca transformar a forma como as pessoas vivenciam a mobilidade pessoal por meio de veículos totalmente autônomos projetados em torno das necessidades dos passageiros.
Gerente de Engenharia, Otimização de Desempenho de ML na Zoox
Lidere a otimização do desempenho de machine learning para os sistemas de direção autônoma da Zoox. Oriente treinamento escalável, aceleração por GPU e inferência de baixa latência entre equipes de engenharia de robótica.
Estados Unidos
Califórnia$ 276.000 – $ 343.000 / ano
HíbridoTempo integralSêniorLíderIdioma Obrigatório: inglêsGerente de engenhariaDescrição
- Definir a direção estratégica e o roteiro para a otimização do treinamento e da inferência de machine learning
- Melhorar a escalabilidade, a confiabilidade e o desempenho das cargas de trabalho de direção autônoma
- Liderar uma plataforma de ML eficiente que abranja treinamento, validação, serving, otimização e monitoramento de modelos
- Otimizar o treinamento e a inferência de modelos em larga escala de ponta a ponta
- Aumentar a eficiência do treinamento distribuído, melhorando a utilização de GPU, o uso de memória e a comunicação
- Usar quantização, poda, destilação e outros métodos de compressão de modelos
- Entregar inferência de baixa latência no veículo dentro de restrições rigorosas de tempo real e capacidade computacional
- Formar e motivar uma equipe de engenharia diversificada por meio de contratação e liderança
- Colaborar com equipes de pesquisa de ML e de engenharia de software, dados e hardware
- Estabelecer requisitos e orientar decisões arquiteturais
- Orientar engenheiros por meio de oportunidades de desenvolvimento e feedback oportuno
Requisitos
- Pelo menos 8 anos de experiência relevante, incluindo 3 anos ou mais gerenciando equipes de engenharia
- Profundo conhecimento especializado em otimização de desempenho de ML, incluindo treinamento distribuído, paralelismo de dados, tensores e pipeline, FSDP/ZeRO, precisão mista, CUDA, Triton, torch.compile, XLA, TVM, quantização e criação de perfis em aceleradores GPU e embarcados
- Experiência na criação de infraestrutura de ML acessível para treinamento em larga escala e serving de alto throughput e baixa latência
- Experiência usando PyTorch ou JAX com GPUs para treinamento distribuído de modelos
- Experiência com inferência acelerada por GPU usando TensorRT, Ray Serve ou frameworks comparáveis
- Histórico sólido de trabalho multifuncional com equipes de pesquisa, produto, hardware e plataforma
- Capacidade de definir prioridades, orientar a direção técnica e entregar ganhos de desempenho mensuráveis entre diferentes áreas da organização
Benefícios
- Licença remunerada para doença, férias e luto
- Licença não remunerada
- Direitos de valorização de ações da Zoox
- Unidades de ações restritas da Amazon (RSUs)
- Seguro-saúde
- Seguro de cuidados de longo prazo
- Seguro de invalidez de longo e curto prazo
- Seguro de vida
- Possível bônus de contratação