Software Engineer II, Behavior Planning ML Platform

Aurora is a company dedicated to delivering the benefits of self-driving technology. They are seeking a Software Engineer to join their Behavior Planning ML Platform team, responsible for building the infrastructure for data extraction, model training, and evaluation, as well as deploying models onboard vehicles.

Responsibilities

• 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

Skills

• 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
• 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

Benefits

• Annual bonus
• Equity compensation
• Benefits

Company Overview

Company H1B Sponsorship