Job Description
The ML Engineering Team Lead at Aurora will develop and deploy advanced machine learning models for autonomous vehicle behavior planning, lead technical initiatives, and guide a team to push the boundaries of self-driving technology in real-world and simulated environments.
Key Responsibilities
- Define the architecture of onboard planning models for autonomous vehicles
- Develop and deploy large-scale models trained with Imitation Learning and Reinforcement Learning
- Build and enhance simulation engines and realistic world models for testing autonomous driving systems
- Create offboard critic models to evaluate driving behavior at scale
- Push forward research to advance autonomous driving technology and deploy models in real-world vehicles
- Mentor and lead engineering teams, guiding the development of ML-based planning systems
Requirements
- MS or PhD in Robotics, Machine Learning, Computer Science, or a related quantitative field, or equivalent practical experience.
- 8 years of experience developing state-of-the-art ML models, either in a research or production setting.
- Hands-on experience working on Imitation Learning or Reinforcement Learning applied to physical or simulated agents.
- Experience training large models on massive datasets using distributed computing.
- Fluency in Python, with a focus on writing high-performance, maintainable code.
- Deep experience with PyTorch preferred or another modern ML framework, and a mastery of modern ML architectures including Transformers and Diffusion Models.
- Ability to define the architecture of onboard planning models.
- Experience developing and deploying large-scale models trained with Imitation Learning and Reinforcement Learning that enable autonomous navigation in complex environments.
- Experience building simulation engines and architecting offboard foundation models that create realistic world models for testing autonomous driving systems.
- Experience developing powerful offboard critic models that evaluate driving behavior at scale, identifying nuances in comfort, progress, and safety.
- Ability to deploy ML models on real production vehicles that drive on public roads and meet the highest safety standards.
- Ability to serve as a technical lead, guiding junior engineers and shaping the long-term ML roadmap for onboard and offboard systems.
Benefits & Perks
Compensation/salary range: 126K - 181.5K per year
Annual bonus
Equity compensation
Benefits (unspecified)
Ready to Apply?
Join Aurora and make an impact in renewable energy
Stay Updated on Sustainability Jobs
Get the latest renewable energy jobs and career tips delivered to your inbox.
Job Alerts
Get notified about new sustainability jobs
More at Aurora
Jobs in Pittsburgh, Pennsylvania
More jobs at Aurora
Tactical Development Testing Operator - CDL A
Aurora
NEW
Dallas
Full Time
11h
Staff Software Engineer, Continuous Learning
Aurora
San Francisco
Full Time
3d
$189k-303k
Software Engineer, MLDE Labels Platform
Aurora
Seattle
Full Time
3d
$126k-201k
More jobs in Pittsburgh, Pennsylvania
Shift Supervisor
Jushi
Pittsburgh
Full Time
3d
Engineering Manager Water Resources
res
Pittsburgh
Full Time
Feb 20
Plumbing Fire Protection II
Cannon Design
Pittsburgh
Full Time
Jan 31