Waymo

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

Waymo

San Francisco•Remote•7-10 Yrs•Jun 22

Remote

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver.

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.
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

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.
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.

Browse Jobs