Software Engineer, Behavior Planning ML Platform
About The Position
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.
Requirements
- 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.
Nice To Haves
- 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
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.
Benefits
- The successful candidate will also be eligible for an annual bonus, equity compensation, and benefits.
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Mid Level
