Senior Software Engineer, ML Data Infrastructure (AI)

TL;DR

Senior Software Engineer, ML Data Infrastructure (AI): Developing and scaling a planetary-scale data infrastructure platform to enable the ML flywheel for various models, including LLMs, with an accent on data store efficiency and orchestration. Focus on building highly scalable distributed systems, optimizing data sharing across models, and integrating solutions for distillation and active learning.

Location: Mountain View, California (Hybrid/On-site)

Salary: $213,000—$263,000 USD

Company

Waymo is an autonomous driving technology company focused on building the world's most experienced autonomous driver to improve mobility and safety.

What you will do

• Develop and contribute to the data infrastructure platform to enable planet-scale ML flywheels for all ML models.
• Partner with modeling teams to onboard LLMs and other models, analyzing data distributions, quality, and value.
• Improve the efficiency of data storage and data sharing mechanisms across different models.
• Deploy and integrate data solutions for distillation, evaluation, dataset generation, and auto-labeling.

Requirements

• 4+ years of professional software engineering experience.
• Strong programming proficiency in C++.
• Experience building highly scalable distributed systems.
• Must be based in or able to work from Mountain View, California.

Nice to have

• Passion for building ML infrastructure and tools.
• Experience handling exabyte-scale datasets.
• Experience with machine learning infrastructure and model hosting/inference systems.
• Experience with production services handling high QPS.

Culture & Benefits

• Participation in a discretionary annual bonus program.
• Equity incentive plan.
• Generous company benefits program.