Job Description
The ML Engineering Team Lead at Aurora will develop and deploy advanced machine learning models for autonomous vehicle behavior planning, oversee simulation and evaluation systems, and guide a team of engineers to push the boundaries of self-driving technology while ensuring safety and real-world deployment.
Key Responsibilities
- Define the architecture of onboard planning models
- Develop and deploy large-scale models using Imitation Learning and Reinforcement Learning
- Build and enhance the simulation engine and offboard foundation models
- Create offboard critic models to evaluate driving behavior at scale
- Deploy ML models on real vehicles to meet safety standards
- Guide and mentor junior engineers and shape the ML planning roadmap
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 for autonomous vehicles.
- 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 developing offboard foundation models that create realistic world models for testing autonomous systems.
- Experience developing powerful offboard critic models that evaluate driving behavior at scale, identifying nuances in comfort, progress, and safety.
- Experience deploying complex ML systems in production environments that meet safety standards.
- Ability to push forward the state-of-the-art in autonomous driving technology through research and development.
- Ability to serve as a technical lead, guiding junior engineers and shaping the long-term ML roadmap for onboard and offboard systems.
Benefits & Perks
Base salary range of 171K - 247K per year
Annual bonus
Equity compensation
Hybrid work environment with in-office presence at least 3 days per week
Commitment to safety policies
Inclusive hiring practices and accommodations for disabilities
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 jobs at Aurora
Senior Training Program Manager
Aurora
NEW
Pittsburgh
Full Time
23h
$100k-145k
Senior Training Program Manager
Aurora
NEW
Mountain View
Full Time
23h
$111k-161k
Truck Operations Specialist
Aurora
NEW
Houston
Full Time
23h
$0k-0k/hr
More jobs in Pittsburgh, Pennsylvania
Patient Care Consultant
Jushi
NEW
Pittsburgh
Full Time
23h
Mechanical - Entry Level PITT
Cannon Design
NEW
Pittsburgh
Full Time
2d
Project Manager II - Ecological Restoration
res
Pittsburgh
Full Time
Mar 10