Software Engineer, Behavior Planning ML Platform
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
Company H1B Sponsorship