Aurora · 5 hours ago
Machine Learning Engineering TL, Behavior Planning
Mountain View, CA
Full-time
Onsite
Senior Level, Lead/Staff
$126K/yr - $182K/yr
8+ years expAurora is dedicated to delivering self-driving technology safely and efficiently. They are seeking a Machine Learning Engineering TL to lead the Behavior Planning Team, focusing on advancing autonomous vehicle technology through innovative ML models and mentoring junior engineers.
AutomotiveAutonomous VehiclesElectric VehicleTransportation
Responsibilities
Define the architecture of our onboard planning models: Develop and deploy large-scale models trained with Imitation Learning and Reinforcement Learning that enable the Aurora Driver to navigate complex environments with human-like fluidity and superhuman safety
Build our next-generation simulation engine: Architect cutting-edge offboard foundation models that power our simulation engine, creating realistic "world models" to test the Aurora Driver against an infinite variety of edge cases
Revolutionize evaluation: Develop powerful offboard critic models that can evaluate driving behavior at scale, identifying subtle nuances in comfort, progress, and safety that traditional heuristics miss
Bridge research and production: Reach new frontiers of autonomous driving technology by pushing forward the state-of-the-art, but also deploy your models on real production vehicles that drive on public roads and must meet the highest standards of safety
Mentor and lead: Serve as a technical lead, guiding junior engineers and shaping the long-term roadmap for ML-based planning at Aurora, including the onboard and off-board ecosystem that is needed to support it
Qualification
Required
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
Preferred
A track record of publications in top-tier ML conferences (NeurIPS, ICML, CoRL, CVPR, AAAI)
Experience deploying complex ML systems in production environments
Experience in developing generative models or neural simulators for synthetic data generation
Experience leading small or large teams to execute highly technical projects
Benefits
Annual bonus
Equity compensation
Benefits
Company
Aurora
Aurora is building self-driving technology to operate multiple vehicle types, from freight-hauling trucks to ride-hailing passenger ones.
1001-5000 employees
H1B Sponsorship
Aurora has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (11)
2024 (51)
2023 (91)
2022 (77)
2021 (38)
2020 (51)
Funding
Current Stage
Public CompanyTotal Funding
$4.19BKey Investors
UberSequoia Capital
2024-08-01Post Ipo Equity· $483M
2023-07-19Post Ipo Equity· $820M
2021-11-04Post Ipo Equity· $1.8B
Leadership Team
Chris Urmson
CEO and co-founder
David Maday
Chief Financial Officer
Recent News
2025-12-24
Company data provided by crunchbase