Senior Software Engineer II, Distributed Training

Key Skills

Algorithms, C++, Machine learning techniques, Optimization, Python Programming, PyTorch, Software Development, TensorFlow

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.

We’re Cruise, the self-driving ride-hailing service.

We are building the world’s most advanced, all-electric, self-driving cars to safely connect people to the places, things, and experiences they care about. We believe self-driving cars will help save lives, reimagine cities, redefine time in transit, and restore freedom of movement for many.

At Cruise, our engineers have opportunities to grow and develop while learning from leaders at the cutting-edge of their fields. With a culture of internal mobility, there's opportunity to thrive in a variety of disciplines. This is a place for dreamers and doers to succeed.

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.

About the role:

The Training Framework team builds and supports the essential tools and frameworks for every machine learning engineer at Cruise. Our goal is to greatly accelerate the development cycle of machine learning models across the whole company, empowering machine learning engineers to focus on improving the car’s safety and performance, instead of worrying about their infrastructure. We care about performance, ease of use and reliability of our products. Our products enable fast, efficient and reliable model training, evaluation and online inference using flexible data sources relevant to Cruise AV and Cruise App. It runs on top of scalable and reliable compute (orchestration & substrate) systems.

Day-to-day responsibilities include:

• Design, implement and deploy platforms and tools to support machine learning models training/evaluation workflows at Cruise.
• Own technical projects from start to finish and be responsible for major technical decisions and tradeoffs. Effectively participate in team’s planning, code reviews and design discussions.
• Consider the effects of projects across multiple teams and proactively manage conflicts. Work closely with partner teams to ensure they are benefiting from the systems we built.
• Conduct technical interviews with well-calibrated standards and play an essential role in recruiting activities. Effectively onboard and mentor junior engineers and/or interns.

You should apply for this role if you have the following qualifications:

• Strong coding in Python or C++
• Experience with distributed training (PyTorch, Tensorflow, etc)
• Experience building large-scale distributed applications, platforms, framework and/or libraries with high-quality API design
• High Engineering Excellence and Operational Excellence mindset
• Passionate about self-driving technology and its potential impact on the world
• BS, MS or PhD in CS, Math or equivalent real-world experience
• Can do attitude and willingness to code

Bonus points!

• Experience building distributed systems on cloud infrastructure
• Knowledge and experience with machine learning algorithms
• Experience scaling model performance optimization which work across many teams
• Experience in Ray or to other distributed machine learning deployment libraries
• Experience with build systems (Bazel, Buck, Blaze or Cmake)

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