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 Aurora is building self-driving technology to operate multiple vehicle types, from freight-hauling trucks to ride-hailing passenger ones. It was founded in 2017, and is headquartered in Pittsburgh, Pennsylvania, USA, with a workforce of 1001-5000 employees. Its website is Company H1B Sponsorship Aurora has a track record of offering H1B sponsorships, with 4 in 2026, 11 in 2025, 51 in 2024, 88 in 2023, 76 in 2022, 38 in 2021, 51 in 2020. Please note that this does not guarantee sponsorship for this specific role.