Aurora Innovation is dedicated to delivering self-driving technology safely and efficiently. They are seeking a Software Engineer to join their Behavior Planning ML Platform team, responsible for developing large-scale data pipelines and optimizing onboard ML infrastructure for model deployment and inference.
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 https://aurora.tech.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.