Machine Learning Engineer (Model Optimization)

Join Waymo, a leader in autonomous driving technology. As a Machine Learning Engineer, you will optimize model training and inference, analyze real-time inference, and collaborate with research teams. Enjoy a hybrid work model, competitive compensation, and a range of benefits including medical, dental, and vision insurance, parental leave, and educational reimbursement.

• Optimize FLOPs utilization in model training and model inference through model architecture/hardware co-development.
• Analyze and optimize real-time inference of complex model architectures with many model components as well as on the critical path within an onboard system.
• Develop methods for efficiently and continuously learning from large scale real-world data, to develop models and model training at scale.
• Bachelors in Computer Science or a similar discipline, or an equivalent amount of deep learning experience
• Experience with Python
• 3+ years experience in Machine Learning and/or Computer Vision
• Experience with ML frameworks like PyTorch or JAX
• Experience with C++
• Publications at top-tier conferences like CVPR, ICCV, ECCV, ICLR, ICML, ICRA, IROS, RSS, NeurIPS, AAAI, IJCV, PAMI
• MS or PhD Degree in Machine Learning, Robotics, Computer Science or a similar discipline