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.
