Staff ML Engineer, Perception Research at Waymo

Waymo is hiring a Principal Research Scientist to develop and deploy multimodal large language models and world models for 3D perception using camera, LiDAR, and radar data. The role focuses on scalable model training, efficient inference, quantization, pruning, distillation, attention mechanisms, data pipelines, evaluation frameworks, model partitioning, and performance analysis. It requires a PhD or master's degree in a relevant technical field and at least four years of industry or postdoctoral research experience in reinforcement learning or foundation models, along with expertise in distributed training, JAX, Flax, and Transformer optimization. The position follows a hybrid schedule and offers a base salary of $251,000 to $310,000 USD.