Staff Deep Learning Optimization Engineer

Employment Type

: Full-Time


: Engineering

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 dreams 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 AV software stack heavily relies on machine learning techniques to perform a variety of tasks, each with different requirements of hardware/compute resources. Throughout the life-cycle of each machine learning model, skilled ML engineers (on both training and inference sides) work closely to prepare it for a robust, scalable, and compute/power efficient inferencing on a resource-constrained hardware accelerator. Such a close working relationship is key to a fast and successful deployment of intelligent systems on the car.

Cruise is looking for ML optimization engineers to help us invest in tools to optimize deep learning models for efficient inference. These optimizations are designed to reduce latency, improve throughput, reduce memory footprint without sacrificing model accuracy. Such techniques may include (but is not limited to) model quantization, model compression and graph manipulation.

In this role, you will collaborate closely with engineers from different AV Engineering teams (e.g. Computer Vision, Perception, platform) to scope out system/software requirements while working with AV hardware to understand optimization opportunities that are specific to the target hardware platform and its constraints. If you are interested in solving some of the most exciting machine learning optimization problems in the industry, then chat with us!

Day-to-day responsibilities include:

  • Define and lead the team’s technical vision and roadmap for DL model optimizations
  • Research, prototype and survey different model optimization techniques (e.g., compression, quantization) and algorithms that are the current state-of-art.
  • Develop proof-of-concepts of customized optimizations that demonstrate the benefit of your optimizations on real-world models using real-world datasets.
  • Seamlessly incorporate model optimization into existing model training and post-training phases of the Cruise AI workflow
  • A main driver for engineering excellence, setting standard and best practices along with mechanisms for their teams to follow.
  • Drive adoption of new technology by collaborating with other teams within Cruise AI (e.g., perception and ML infrastructure).
  • Contribute to strategic hiring across the company. Able to become a Bar Raiser continuously raising the hiring bar for Cruise.
  • Provide mentorship and leverage the team’s fullest potential to achieve ambitious technical goals.
  • Build and deploy production software at the highest levels of quality and performance
  • Collaborate with cross functional agile teams of AV engineers 
  • You should apply for this role if you have the following qualifications:

  • MS, or higher degree in CS/CE/EE, or equivalent, in industry experience
  • 10+ years of software engineering experience in delivering high-quality production software
  • 3+ years of experience in the field of machine learning and deep learning
  • Strong expertise with DL frameworks like Tensorflow, PyTorch etc.
  • Strong programming skills in Python or C++
  • Experience in shipping model optimization techniques like quantization, NAS, sparsity, hyperparameter tuning
  • Comfortable and experienced in software development lifecycle - coding, debugging, optimization, testing, integration
  • Perks of being a Cruiser

    While doing meaningful work is rewarding in itself, we also offer the following programs and benefits to support the extraordinary humans who serve as the backbone in making our self-driving cars go:

  • Competitive salary and benefits including 401(k) Cruise matching program to help you meet your long-term financial goals
  • Medical / dental / vision, AD+D and Life
  • Flexible vacation and company paid holidays
  • Healthy lunch, snacks, dinner
  • Paid parental leave & family expansion stipend
  • Monthly gym stipend
  • Quarterly offsites & working retreats
  • On-site massages
  • State-of-the-art equipment for your work station
  • * The salary listed in the header is an estimate based on salary data for similar jobs in the same area. Salary or compensation data found in the job description is accurate.

    Launch your career - Create your profile now!

    Create your Profile

    Loading some great jobs for you...