Post a job

Staff Machine Learning Engineer, ML Compute


United States
Base Salary
230k-270k USD
Cruise LLC

Job Description

We're Cruise, a self-driving service designed for the cities we love.

We’re building the world’s most advanced self-driving vehicles to safely connect people to the places, things, and experiences they care about. We believe self-driving vehicles will help save lives, reshape cities, give back time in transit, and restore freedom of movement for many.

In our cars, you’re free to be yourself. It’s the same here at Cruise. We’re creating a culture that values the experiences and contributions of all of the unique individuals who collectively make up Cruise, so that every employee can do their best work.

Cruise is committed to building a diverse, equitable, and inclusive environment, both in our workplace and in our products. If you are looking to play a part in making a positive impact in the world by advancing the revolutionary work of self-driving cars, come join us. Even if you might not meet every requirement, we strongly encourage you to apply. You might just be the right candidate for us.

(Additional locations: San Francisco CA, Sunnyvale CA)

The ML Compute Platform is part of Cruise’s AI Foundation org and owns cloud-agnostic, reliable, and cost-efficient compute backend for the specific needs of Cruise AI. We enable high agility for building new features, supporting innovation, and optimizing for prioritized use cases that are relevant to Cruise AI/ML today, and in the future. This team is tasked with ensuring the reliable training and deployment of SOTA ML models and associated workloads, prioritizing high performance, concurrency, availability, and scalability. We emphasize enhancing efficiency in both ML training and deploying cutting-edge foundational models, along with maximizing the utilization of powerful GPUs such as H100, A100 and more. Our team objectives revolve around reliability, cost-effectiveness, scalability, and velocity.

We are looking for an ML Engineer specializing in constructing and managing robust, scalable, and high-performance compute platforms tailored for ML workflows. Collaborating closely with our scientists, they will ensure efficient model training and seamless integration into our production environment, ultimately ensuring the safe and continuous operation of Cruise and our fleet of autonomous vehicles. The ideal candidate will have experience building and running scalable distributed machine learning platforms, will bring innovative ideas and approaches, should have intellectual curiosity, strong problem-solving skills, and a bias towards action.

If you are looking to solve one of today’s most complex engineering challenges, see the results of your work in hundreds of self-driving cars, and make a positive impact in the world starting in our cities, join us!

What you’ll be doing:

  • Design core platform backend software components

  • Experience cloud platforms like GCP, Azure

  • Thrive in a dynamic, multi-tasking environment with ever-evolving priorities. Interface with other teams to incorporate their innovations and vice versa

  • Analyze and improve efficiency, scalability, and stability of various system resources

  • Proactively identify, drive and design large initiatives across Cruise ML workflows

  • Work on large scale initiatives to raise the overall Cruise engineering bar

  • Participate/lead open source projects. Lead/drive community recognition for Cruise engineering.

At a Minimum We'd Like You To Have

  • 8+ years of industry experience

  • Expertise in either Go, C++, Python or other relevant coding languages

  • Strong background with kubernetes at scale

  • Relevant experience building large-scale with distributed systems

  • Experience leading and driving large scale initiatives

  • Experience working with Google Cloud Platform, Microsoft Azure, or Amazon Web Services

It's Preferred If You Have

  • Hands-on experience in ML platforms

  • Experience with GPU/TPU optimizations

  • Experience with training frameworks like PyTorch, TorchX

  • Experience with Ray framework

  • Leadership/active participation in the open source community

  • Experience infrastructure applications or similar experience

The salary range for this position is $229,500 - $270,000. Compensation will vary depending on location, job-related knowledge, skills, and experience. You may also be offered a bonus, and benefits. These ranges are subject to change.

Why Cruise?

Our benefits are here to support the whole you:

  • 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. (Learn more about CruiseFlex).

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.

Cruise LLC is an equal opportunity employer. We strive to create a supportive and inclusive workplace where contributions are valued and celebrated, and our employees thrive by being themselves and are inspired to do the best work of their lives. We seek applicants of all backgrounds and identities, across race, color, caste, ethnicity, national origin or ancestry, citizenship, religion, sex, sexual orientation, gender identity or expression, veteran status, marital status, pregnancy or parental status, or disability. Applicants will not be discriminated against based on these or other protected categories or social identities. Cruise will consider for employment qualified applicants with arrest and conviction records, in accordance with applicable laws.

Cruise is committed to the full inclusion of all applicants. If reasonable accommodation is needed to participate in the job application or interview process please let our recruiting team know or email [email protected].

We proactively work to design hiring processes that promote equity and inclusion while mitigating bias. To help us track the effectiveness and inclusivity of our recruiting efforts, please consider answering the following demographic questions. Answering these questions is entirely voluntary. Your answers to these questions will not be shared with the hiring decision makers and will not impact the hiring decision in any way. Instead, Cruise will use this information not only to comply with any government reporting obligations but also to track our progress toward meeting our diversity, equity, inclusion, and belonging objectives.

Candidates applying for roles that operate and remotely operate the AV: Licensed to drive a motor vehicle in the U.S. for the three years immediately preceding your application, currently holding an active in-state regular driver’s license or equivalent, and no more than one point on driving record. A successful completion of a background check, drug screen and DMV Motor Vehicle Record check is also required.

Note to Recruitment Agencies: Cruise does not accept unsolicited agency resumes. Furthermore, Cruise does not pay placement fees for candidates submitted by any agency other than its approved partners.

Apply for this job


Please let Cruise LLC know you found this job with This helps us grow! mascot