Principal AI/ML Engineer, AV ML Infra

Description

The company develops self-driving and electric vehicle technology.

The AV ML Infra team builds machine learning infrastructure for autonomous driving work and supports groups such as Embodied AI, Simulation and Data Science.

The platform helps engineers run large-scale experiments more efficiently, raises their productivity and brings newer ML methods into everyday use.

The stack has three parts: AI validation and inference, which runs large simulation jobs and manages reliable inference pipelines; ML compute, which optimizes training and inference across cloud and on-prem resources; and AV pipelines and lineage, which orchestrates ML workflows and tracks data and model lineage for reproducibility.

This leadership role covers a growing organization. You will set long-term direction and execution plans for the company's AI and ML work, lead a major step up in infrastructure capacity for expected scale, design the team structure and build strong ties with cross-functional partners.

The role sits within infrastructure engineering and does not apply ML models to specific tasks. The goal is to build infrastructure products that let company teams do machine learning and data science at scale.

Responsibilities

• Design, build and test scalable distributed computing and data processing systems on current cloud platforms, GCP or Azure.
• Own technical projects from start to finish, help shape the product roadmap and make sound calls on major tradeoffs.
• Take part in planning, code reviews and design discussions, weighing how each project affects other teams and resolving disagreements early.
• Run technical interviews to a consistent bar, onboard engineers and interns, and coach them while encouraging learning and knowledge sharing.

Requirements

• 10+ years of experience, ideally with large-scale distributed systems.
• 5+ years of experience leading large initiatives.
• Skill building scalable cloud infrastructure in Python, C++, Golang or a similar language.
• Experience with relational and NoSQL databases.
• A record of building and maintaining systems at scale.
• A bachelor's, master's or PhD in computer science, electrical engineering, mathematics, physics or a related field, or equivalent practical experience.
• Genuine interest in autonomous vehicle technology and its potential.
• Careful attention to detail and accuracy.
• A history of solving complex problems efficiently.
• A startup mindset: comfort with uncertainty and willingness to take on several roles at once.
• Bonus: experience with Google Cloud Platform, Microsoft Azure or Amazon Web Services.
• Bonus: experience with open-source orchestration platforms such as Kubeflow, Flyte or Airflow.
• Bonus: experience with model serving frameworks such as RayServe, vLLM or Triton.
• Bonus: experience with Kubernetes.
• Bonus: understanding of ML models and pipelines.
• Bonus: proficiency in Python, C++ or Golang.
• Bonus: relevant publications.

Conditions

Base salary falls between $275,800 and $340,500, with the final offer set by relevant factors within that range.

This figure is a good-faith estimate under applicable state law and may not reflect roles outside New York, Colorado, California or Washington.

Incentive pay depends on company performance, job level and individual performance.

The role is remote-based, but if you live within 50 miles of Mountain View, you must report to that office at least three days a week.

This job may be eligible for relocation benefits.

You may need to complete a role-related assessment and/or pre-employment screening before you start.

The company is committed to an inclusive workplace free of unlawful discrimination.

Benefits

• Health and wellbeing programs including medical, dental, vision, a Health Savings Account and Flexible Spending Accounts.
• Retirement savings plan, sickness and accident benefits, life insurance and paid vacation and holidays.
• Tuition assistance, an employee assistance program, company vehicle discounts and other benefits.