Software Engineer II, Behavior Planning ML Platform

Who we are

Aurora's mission is to deliver the benefits of self-driving technology safely, quickly, and broadly.

The Aurora Driver will create a new era in mobility and logistics, one that will bring a safer, more efficient, and more accessible future to everyone.

At Aurora, you will tackle massively complex problems alongside other passionate, intelligent individuals, growing as an expert while expanding your knowledge. For the latest news from Aurora, visit aurora.tech or follow us on LinkedIn.

Aurora hires talented people with diverse backgrounds who are ready to help build a transportation ecosystem that will make our roads safer, get crucial goods where they need to go, and make mobility more efficient and accessible for all. We're searching for a Software Engineer to join Aurora's Behavior Planning ML Platform team. Our team is responsible for building both the offline pipelines to train and evaluate models and the infrastructure to efficiently run these models onboard the vehicle. On the offboard side, we build and support the large scale distributed pipelines that model developers rely on to source data at scale, train, and evaluate machine learning models. On the onboard side, we build and maintain the on-vehicle infrastructure required to deploy these models and run inference in a soft real-time system.

In this role, you will

• Develop large scale pipelines for data extraction, model training and model evaluation
• Build and optimize onboard ML infrastructure used to deploy models and run inference onboard the vehicle
• Collaborate closely with motion planning, systems engineering, and other autonomy groups to define and develop critical ML workflow requirements.

Required Qualifications

• BS or higher degree in Computer Science/Engineering or related fields. > 6 months of experience
• Strong programming skills in C++ or Python, ideally both
• Experience with machine learning frameworks (PyTorch or TensorFlow)
• Solid foundation in computer science fundamentals - especially operating system concepts including concurrency, memory management and process scheduling.

Desirable Qualifications (this section is optional)

• Experience with GPU programming and Nvidia profiling tools - CUDA, NSight Systems, Nsight Compute
• Experience working in cloud compute (AWS, Azure, Google Cloud Platform) ideally with batch processing or other large-scale workflow management platforms.
• Experience profiling and optimizing software on Linux

The base salary range for this position is $126,000 - $181,000 per year. Aurora's pay ranges are determined by role, level, and location. Within the range, the successful candidate's starting base pay will be determined based on factors including job-related skills, experience, qualifications, relevant education or training, and market conditions. These ranges may be modified in the future. The successful candidate will also be eligible for an annual bonus, equity compensation, and benefits.

#LI-JL261

#Mid-Senior

Working at AuroraAt Aurora, we bring together extraordinarily talented and experienced people united by the strength of our values. We operate with integrity, set outrageous goals, and build a culture where we win together - all without any jerks.

We believe in-person work increases collaboration, empathy and our ability to lead effectively. As a result, we operate in a hybrid work environment where Aurorans are in office at least 3 days per week.

Our Careers page provides insight into what it is like to work at Aurora, and you can find all the latest updates in our Newsroom.

Our commitment to safety

At the core of everything we do is our commitment to safety. Building best-in-class self-driving technology will take time, and we believe that each employee at Aurora has a role in contributing to safety, every step of the way. Aurora expects commitment to our safety policies from every employee, and seeks candidates who take an active responsibility, can contribute to building an atmosphere of trust, and invest in the organization's long-term success by prioritizing working safely, no matter what.

Our commitment to inclusion

Aurora considers candidates without regard to their race, color, religion, national origin, age, sex, gender, gender identity, gender expression, sexual orientation, marital status, pregnancy status, parent or caregiver status, ancestry, political affiliation, veteran and/or military status, physical or mental disability, or any other status protected by federal or state law. Aurora considers qualified applicants with criminal histories, consistent with applicable federal, state, and local law. We are also committed to providing reasonable accommodations for qualified individuals with disabilities and disabled veterans in our job application procedures. If you need assistance or an accommodation due to a disability, you may contact us at

For California applicants, information collected and processed as part of your application and any job applications you choose to submit is subject to Aurora's California Employment Privacy Policy.