Software Engineer II, Behavior Planning ML Platform

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, GCP) 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