Senior Machine Learning Engineer (Infra, Driver Understanding and Evaluation)

Join Waymo, a leader in autonomous vehicle technology, as a Senior Machine Learning Engineer. In this role, you will build and operate scalable machine learning and data systems, improve evaluation processes, and contribute to the production and optimization of machine learning models. You will work at the intersection of data engineering, model development, and simulation, and collaborate cross-functionally to derive performance and system-level requirements. This position offers a competitive compensation package, comprehensive benefits, and opportunities for professional growth.

Contribuer à la production et à l'optimisation de modèles d'apprentissage automatique visant à évaluer la flotte de véhicules de Waymo.

We are looking for researchers and software engineers who are passionate about developing machine learning techniques for the Evaluation systems on our autonomous vehicles, and have an incessant drive to improve the performance of our technology stack

About Waymo

• Concevoir et mettre à l'échelle des systèmes distribués couvrant le cycle de vie de l'apprentissage automatique, soutenant la génération de jeux de données à l'échelle planétaire.
• Collaborer de manière transversale pour dériver les performances et les exigences au niveau système pour les grands systèmes d'apprentissage automatique.

Responsibilities

• Strong expertise in distributed training techniques, including gradient sharding and optimization strategies for scaling large models across ML accelerator profiling tools to uncover performance bottlenecks

Requirements

• A history of contributions to machine learning tooling and frameworks e.g. PyTorch, Jax, Tensorflow, Ray, or similar. The candidate should understand both the user facing API and the internal workings
• 5+ years in machine learning infrastructure such as developing, designing, scaling, training, deploying, and optimizing large-scale machine learning systems from data to model
• M.S. or Ph.D. degree Computer Science, Machine Learning, Artificial Intelligence, or a related technical field, or equivalent practical experience
• Experience in the autonomous vehicles domain, robotics, or complex simulation environments
• 7+ years in machine learning infrastructure such as developing, designing, scaling, training, deploying, and optimizing large-scale machine learning systems from data to model

Preferred

• Familiarity with large-scale simulation platforms and their integration with ML training workflows

Pay

USD 213,000 - 263,000