Waymo

Senior Staff Machine Learning Engineer, LLM/VLM Model Architecture & Optimization

Waymo$298K — $368K *
Consumer Technology
5 - 7 years of experience
Job Overview by Ladders

Qualifications

  • 7+ years of experience in Machine Learning, focusing on LLM and VLM development.
  • Expertise in low-latency on-device inference and hardware acceleration.
  • Proficient in deep learning frameworks like PyTorch and JAX.
  • Ability to work effectively in ambiguous situations and establish direction.
  • Experience in applying large language models in safety-critical domains like robotics or autonomy.
  • Master's degree in Computer Science, Electrical Engineering, or a related field, or equivalent experience.

Responsibilities

  • Design model architectures aligning with hardware requirements.
  • Optimize models for on-device performance under constraints.
  • Collaborate with cross-functional teams for end-to-end solutions.

Benefits

  • Participation in an annual bonus program.
  • Eligibility for an equity incentive plan.
  • Access to a generous company benefits program.
Full Job Description


You will:
  • 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.

You have:
  • 7+ 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.
  • 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.

We prefer:
  • Familiarity with large-scale data curation and quality assurance processes for multimodal datasets.
  • Background in autonomous vehicle perception, motion planning, or decision-making systems.
  • Publications in top-tier machine learning or computer vision conferences (e.g., NeurIPS, ICML, CVPR, ICCV, ECCV).
  • PhD in a relevant field.


The expected base salary range for this full-time position across US locations is listed below. Actual starting pay will be based on job-related factors, including exact work location, experience, relevant training and education, and skill level. Your recruiter can share more about the specific salary range for the role location or, if the role can be performed remote, the specific salary range for your preferred location, during the hiring process.

Waymo employees are also eligible to participate in Waymo's discretionary annual bonus program, equity incentive plan, and generous Company benefits program, subject to eligibility requirements.

Salary Range

$298,000-$368,000 USD

About Waymo

Waymo LLC is a self-driving technology development company. It is a subsidiary of Alphabet Inc., the parent company of Google. Waymo develops autonomous driving technology and provides ride-hailing services through its Waymo One program. The company has been testing its self-driving technology on public roads since 2009 and has logged millions of miles of autonomous driving. Waymo has partnerships with several automakers and has been working on developing autonomous trucks for use in logistics.
Learn more about Waymo
Size
2,000 employees
Industry
Founded
2009

Similar Jobs

More Jobs at Waymo

More Consumer Technology Jobs

Find similar Senior Staff Machine Learning Engineer, LLM/VLM Model Architecture & Optimization jobs: