Staff Machine Learning Engineer (Tech Lead, Labeling Automation)

Join Waymo as a Staff Machine Learning Engineer (Tech Lead, Labeling Automation). In this high-impact leadership role, you will lead the technical strategy for automating data pipelines, build cutting-edge auto-labeling systems, and develop intelligent auto-graders to ensure exceptional data quality. You will collaborate cross-functionally with various teams and drive the technical direction for the Labeling organization. Enjoy a comprehensive benefits package, including medical, dental, and vision insurance, competitive compensation, and a hybrid work model.

• Lead the design and deployment of highly scalable auto-labeling pipelines that significantly improve data throughput and reduce reliance on manual annotation.
• Develop automated anomaly detection and quality evaluation systems (auto-graders) to assess annotation accuracy, detect regressions, and enforce rigorous quality standards.
• Train, optimize, and push into production advanced 2D and 3D computer vision models, utilizing architectures ranging from foundational zero-shot models to bespoke 3D perception models.
• Experience programming in C++ or Python
• Experience taking machine learning solutions through the entire lifecycle—from research and experimentation to robust, scaled production deployment
• 8+ years of professional experience in the field of software engineering and applied machine learning
• Experience in large model training, distributed computing, and scaling deep learning architectures using frameworks like PyTorch or TensorFlow
• Experience building, evaluating, and deploying deep learning models for object detection, segmentation, and spatial tracking
• Experience building internal tooling for ML developers