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 jobs at Aurora
Data Platform Engineer
Aurora
NEW
Pittsburgh
Full Time
2d
$105k-157k
Security Engineering Technical Lead Manager TLM - Aurora Enterprise Security
Aurora
Seattle
Full Time
3d
$189k-274k
FP A Analyst Procurement Systems
Aurora
Pittsburgh
Full Time
4d
$104k-166k
More jobs in Pittsburgh, Pennsylvania
Patient Care Consultant
Jushi
Pittsburgh
Full Time
Nov 3
Patient Care Consultant
Jushi
Pittsburgh
Full Time
Nov 17
Shift Supervisor
Jushi
Pittsburgh
Full Time
Nov 18