Senior Staff Machine Learning Engineer, LLM/VLM Model Architecture & OptimizationPosted today$368K

The opportunity

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver™—to improve access to…

What you'll do

  • Design VLM/LLM model architecture and drive strong alignment between model: architectures and hardware architectures.

  • Optimize model performance for on-device use cases (memory, power, compute constrained environments).

  • Engage directly with research, software engineering, hardware engineering,: and product teams to deliver end-to-end solutions.

  • + years of experience in Machine Learning, with a focus on large-scale model: development (LLM, VLM, or similar foundation models).

  • Proven expertise in low-latency on-device inference techniques and a deep: understanding of hardware acceleration.

  • Extensive experience with deep learning frameworks (e.g. PyTorch, JAX) and large-scale model training.

What they're looking for

  • A track record of operating effectively under ambiguity, setting direction: amid rapidly evolving research and technical constraints
  • Experience applying large language models or foundation models in complex,: safety-critical domains (e.g., autonomy, robotics, or other high-reliability systems)
  • Master's degree in Computer Science, Electrical Engineering, or a related: field, or equivalent practical experience.
  • Familiarity with large-scale data curation and quality assurance processes for multimodal datasets.