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

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. The Senior Staff Machine Learning Engineer will design VLM/LLM model architecture, optimize model performance for on-device use cases, and engage with various teams to deliver end-to-end solutions.

Responsibilities:

• 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