Waymo

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

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

Qualifications

  • 7+ years in Machine Learning with focus on large-scale model development (LLM, VLM, or similar)
  • Expertise in low-latency on-device inference and hardware acceleration
  • Experience with deep learning frameworks (e.g. PyTorch, JAX)
  • Capability to operate under ambiguity and quickly adapt to changing conditions
  • Experience in applying large language models in high-reliability systems
  • Master's degree in Computer Science, Electrical Engineering, or related field, or equivalent practical experience

Responsibilities

  • Design VLM/LLM model architecture in alignment with hardware architectures
  • Optimize model performance for on-device, resource-constrained environments
  • Collaborate with research, software, hardware engineering, and product teams for end-to-end solutions

Benefits

  • Eligibility for annual bonus program
  • Participation in equity incentive plan
  • Comprehensive 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 Transportation Jobs

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