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

Highlights

• Build spatial-temporal perception that lets the Waymo Driver see the world.
• Design LLM/VLM architectures aligned with hardware for on-device use.
• Optimize models for memory, power and compute constrained environments.

What they offer

• Discretionary annual bonus program
• Equity incentive plan
• Generous Company benefits program

Ideal candidate

A seasoned ML innovator with 7+ years building foundation models who thrives in ambiguity and cares about deploying safe, low-latency autonomy at scale.

Requirements

• 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
• 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
• Master's degree in Computer Science, Electrical Engineering, or related field, or equivalent practical experience