Senior Staff Machine Learning Engineer, AI Foundations

At Cruise, the AI Foundations organization (AIF) builds the essential AI infrastructure used by every ML engineer at Cruise. AIF's goal is to support the AI lifecycle through the ML data platform, the ML training platform as well as deploying large complex ML models to the on-vehicle hardware under tight real-time constraints. AIF's mission is to provide the highest levels of reliability, the fastest developer velocity at the lowest cost footprint through these platforms and services.

Cruise is building large foundational models that are customized to solve the autonomous driving challenges. This naturally implies that the ML systems and training infrastructure must scale to the requirements of these large models. Deploying these large models to edge (on-vehicle) hardware is another critical challenge. In this role, as a Senior Staff engineer on this team, you will be leading the vision/strategy as well as tech-leading the execution to solve these hard problems. Cruise is a highly collaborative and dynamic work environment. In this role, as an uber TL, you will be responsible for collaboratively working with partners on the ML teams (Autonomy) to inspire a technical strategy / vision and drive execution through a cross-functional team.

What you'll be doing:

• Be a strategic thought leader for AI Foundations as a whole
• Lead strategic execution around ML systems and training infrastructure and deploying large models
• Be as a technical leader across the broader Cruise AI and Engineering organizations to deliver production systems and mentor Engineers
• Define execution processes that strive for streamlined engineering development with robust quality and excellence
• Contribute broadly on the needs and strategy from data, to training, to deployment
• Utilize your deep understanding of data characteristics, model architectures, optimization techniques, or other ML domain-specific challenges to perform critical analysis of modeling results
• Propose ways to improve the quality of the data for the model training and evaluation, automate data collection for reproducibility
• Closely follow industry and academic developments in the SOTA for the AI life-cycle and ML Systems domain's and adopt technology that is the best fit for Cruise's needs
• Own technical projects from start to finish, contribute to the team's product roadmap, and be responsible for major technical decisions and tradeoffs
• Consider the effects of projects across multiple teams and proactively manage risks/conflicts
• Conduct technical interviews and play an essential role in recruiting activities

What you must have:

• 5+ years working on ML systems for training and deploying complex ML models by employing SOTA techniques in the area of distributed training and deployment
• Experience working on the scaling challenges associated with building LLMs and/or other large foundational models
• Passionate about self-driving technology and its potential impact on the world
• Attention to detail and a passion for truth
• A track record of efficiently solving complex problems collaboratively on larger teams
• Startup mentality - openness to dealing with unknown unknowns and wearing many hats
• Strong expertise in writing production quality code and setting standards for code quality across engineering teams
• Experience in driving technical strategy and vision for engineering teams and organizations
• Leadership experience for planning and execution of cross-functional initiatives and projects
• In-depth understanding of software development lifecycle (SDLC) and best practices
• PhD in CS/CE/EE, or equivalent, in industry experience

Bonus points:

• Worked on distilling large foundational models to production on edge devices
• Strong experience in working with GPUs and GPU based datacenters
• Relevant publications

Why Cruise?

• Competitive salary and benefits
• Medical / dental / vision, Life and AD&D
• Subsidized mental health benefits
• Paid time off and holidays
• Paid parental, medical, family care, and military leave of absence
• 401(k) Cruise matching program
• Fertility benefits
• Dependent Care Flexible Spending Account
• Flexible Spending Account & Health Saving Account
• Perks Wallet program for benefits/perks
• Pre-tax Commuter benefit plan for local employees
• CruiseFlex, our location-flexible work policy

We're Integrated: Through our partnerships with General Motors and Honda, we are the only self-driving company with fully integrated manufacturing at scale.

We're Funded: GM, Honda, Microsoft, T. Rowe Price, and Walmart have invested billions in Cruise. Their backing for our technology demonstrates their confidence in our progress, team, and vision and makes us one of the leading autonomous vehicle organizations in the industry. Our deep resources greatly accelerate our operating speed.