Principal ML Systems Architect – Foundation Models

Tech Stack

Responsibilities

• Define and drive the technical roadmap for the platform, encompassing codebase unification, data pipelines, model architecture, training recipes, and evaluation frameworks.
• Lead the unification of existing forked locations of foundation model component codebases into a production-hardened, shared repository.
• Serve as the primary technical interface between Waymo's offboard model development and Google Deepmind's core model and framework teams.
• Drive the consolidation of tokenization, data formats, input pipelines, and evaluation methodologies across all offboard Foundation Model use cases.
• Architect for efficient large-scale distributed training and establish a common, efficient distillation setup to transfer knowledge from large teacher models to onboard student models.

Soft Skills

Benefits

• Equity
• Hybrid Work

Culture

Requirements