Machine Learning Engineering TL, Behavior Planning

$171k - $247k

Full-time

Aurora

Who We Are
Aurora’s mission is to deliver the benefits of self-driving technology safely, quickly, and broadly.
The Aurora Driver will create a new era in mobility and logistics, one that will bring a safer, more efficient, and more accessible future to everyone.
At Aurora, you will tackle massively complex problems alongside other passionate, intelligent individuals, growing as an expert while expanding your knowledge. For the latest news from Aurora, visit aurora.tech or follow us on LinkedIn.
Aurora hires talented people with diverse backgrounds who are ready to help build a transportation ecosystem that will make our roads safer, get crucial goods where they need to go, and make mobility more efficient and accessible for all. We are seeking a ML Engineering TL to join the Behavior Planning Team . This is a high-impact role where you will sit at the frontier of autonomous vehicle technology. You won't just be applying AI and ML methods; you will be advancing the state-of-the-art for how a self-driving system reasons about the world, interacts with humans on the road, and masters complex, high-dimensional decision-making.
In this role, you will
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.
Required Qualifications
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.
Desirable Qualifications
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.
The base salary range for this position is $171K - $247K per year. Aurora’s pay ranges are determined by role, level, and location. Within the range, the successful candidate’s starting base pay will be determined based on factors including job-related skills, experience, qualifications, relevant education or training, and market conditions. These ranges may be modified in the future. The successful candidate will also be eligible for an annual bonus, equity compensation, and benefits.
#Mid-Senior
Working at Aurora
At Aurora, we bring together extraordinarily talented and experienced people united by the strength of our values. We operate with integrity, set outrageous goals, and build a culture where we win together — all without any jerks.
We believe in-person work increases collaboration, empathy and our ability to lead effectively. As a result, we operate in a hybrid work environment where Aurorans are in office at least 3 days per week.
Our Careers page provides insight into what it is like to work at Aurora, and you can find all the latest updates in our Newsroom.
Our commitment to safety
At the core of everything we do is our commitment to safety. Building best-in-class self-driving technology will take time, and we believe that each employee at Aurora has a role in contributing to safety, every step of the way. Aurora expects commitment to our safety policies from every employee, and seeks candidates who take an active responsibility, can contribute to building an atmosphere of trust, and invest in the organization’s long-term success by prioritizing working safely, no matter what.
Our commitment to inclusion
Aurora considers candidates without regard to their race, color, religion, national origin, age, sex, gender, gender identity, gender expression, sexual orientation, marital status, pregnancy status, parent or caregiver status, ancestry, political affiliation, veteran and/or military status, physical or mental disability, or any other status protected by federal or state law. Aurora considers qualified applicants with criminal histories, consistent with applicable federal, state, and local law. We are also committed to providing reasonable accommodations for qualified individuals with disabilities and disabled veterans in our job application procedures. If you need assistance or an accommodation due to a disability, you may contact us at View email address on ev.careers.
For California applicants, information collected and processed as part of your application and any job applications you choose to submit is subject to Aurora’s California Employment Privacy Policy.

Apply

Vacancy posted a month ago

Similar jobs that could be interesting for youBased on the Machine Learning Engineering TL, Behavior Planning in California vacancy

Machine Learning Engineering TL, Behavior Planning Mountain View, California
$126k - $181.5k
...Software Engineering Mountain View, California Machine Learning Engineering TL, Behavior Planning Who we are Aurora’s mission is to deliver the benefits of self-driving technology safely, quickly, and broadly. The Aurora Driver will create a new era in mobility...
Suggested
Local area
Australian Competition and Consumer Commission
Mountain View, CA
1 day ago
Senior ML/RL Engineer, Behavior Planning
...Senior ML/RL Engineer, Behavior Planning At Bot Auto, we are revolutionizing the transportation of goods with our cutting-edge autonomous trucks... ...work at the intersection of Multi-Agent Reinforcement Learning (MARL) and safety-critical system design to ensure our autonomous...
Suggested
Shift work
Bot Auto
San Francisco, CA
3 days ago
Machine Learning Engineer: Evaluation
...construction veterans and world-class engineers to solve physical-world... ...to have you join us. Machine Learning Engineer: Evaluation... ...Connect product goals and system behavior - by bridging real-world specification... ...parameter tuning to program planning by slicing through the noise...
Suggested
Work at office
Flexible hours
Bedrock Robotics
San Francisco, CA
4 days ago
Machine Learning Engineer Perception LLM/VLM (PhD, New Grad)
$170k - $216k
...Machine Learning Engineer Perception LLM/VLM (PhD, New Grad) Waymo is an autonomous driving technology... ...at scale, to (3) analyze real-world behavior and develop systems for handling the... ...vehicle perception, motion planning, or decision-making systems. Publications...
Suggested
Full time
Remote work
Waymo
Mountain View, CA
2 days ago
Senior Machine Learning Engineer - Scene Intelligence
...exceptional accuracy. As an engineer in the Scene Understanding... ...work with state-of-the-art machine learning models that operate in real-... ...researchers across sensors, planning, and other teams, you'll have... ...that improve driving behavior Partner with cross-functional...
Suggested
Temporary work
Relocation package
Zoox
Foster, CA
17 days ago
Senior Machine Learning Engineer, Data Mining
$172k - $229k
...data mining framework, is the engine that powers this discovery. As a Senior Machine Learning Engineer on the Data Mining team... ...prompting, and goal-directed behavior. Integrate such systems into... ...-thought models, or LLM-based planning. Background in autonomous driving...
Work at office
Remote work
Motional
San Francisco, CA
25 days ago
Associate Machine Learning Engineer
$70k - $100k
...Associate ML Engineer/ Agentic AI Are you interested... ...AI, AI systems that can plan, act, and deliver... ...operations Support agent behavior design, including planning... ...and a passion for learning. You might be a great fit... ...automation, and applied machine learning...
Internship
Flexible hours
XIFIN
San Diego, CA
1 day ago
Senior / Staff Machine Learning Engineer - Perception Attributes
...Description Job Description As a machine learning engineer within the Attributes team in the Perception... ...and enhancing sophisticated behavioral models for various road users, including... ...that enable our Prediction and Planning teams to make intelligent, safe driving...
Temporary work
Relocation package
Zoox
Foster, CA
16 days ago
Senior Machine Learning Engineer - ML Agents and Planning
...Foundation Models for ML Agents and planning, applying them off-vehicle to provide... ...performance. As an ML Agents and Planning Machine Learning Engineer you will work on the bleeding edge of... ...pipelines and models to predict the behavior of other agents in the world and...
Temporary work
Relocation package
Zoox
Foster, CA
1 day ago
Staff Machine Learning Engineer - Foundation Model
$215.28k - $364.32k
...Staff Machine Learning Engineer - Foundation Model Santa Clara, CA XPENG is a leading smart technology... ...-class researchers, perception and planning engineers, and infrastructure experts... ...LiDAR, CAN bus, maps, human driving behaviors, etc.). Research and integrate...
Full time
XPENG
Santa Clara, CA
4 days ago
Software Engineer - Learned Trajectory Machine Learning Engineer
$189k - $270k
...The Prediction & Behavior ML team is responsible for developing machine learning (ML) algorithms that learn and predict behaviors... ...forecasting and motion planning, our team collaborates closely... ...Perception, Simulation, and Systems Engineering teams to accelerate our...
Temporary work
Relocation package
Zoox
Foster, CA
3 days ago
Machine Learning Research Engineer/Scientist
...the things they value most. As a Machine Learning Research Engineer, you will work on the software and algorithms... ...robot fleet to continuously add new behaviors and improve robustness across tasks... ...fundamentals (e.g. controls, planning, state estimation) Comfortable...
Sunday
Redwood City, CA
4 days ago
Staff AI Machine Learning Engineer
...Description The Role We are seeking a Staff Machine Learning Engineer with deep expertise in agentic AI —... ...cycles, iterating on novel agent behaviors, and turning unsolved challenges into... ...to-end agentic workflows that reason, plan, use tools, orchestrate multi-agent collaboration...
Medeloop
San Francisco, CA
17 days ago
Machine Learning Engineer - MLA Perception Offline Driving Intelligence
...navigation in complex environments. As an engineer in the ODIN team, you will develop... .... Collaborate with perception, planning, safety, and systems teams to integrate... ...Qualifications MS or PhD in Computer Science, Machine Learning, or related technical field...
Temporary work
Relocation package
Zoox
Foster, CA
29 days ago
3D Machine Learning Engineer
...only architectures, combining rigorous engineering with learning systems proven in globally deployed... ...Do Design and implement scalable machine learning pipelines for large-scale 3D... ...Models™ raise the bar for perception, planning, localization, and manipulation, with...
Local area
FieldAI
Irvine, CA
25 days ago
Machine Learning Engineer
...Description About Machina Labs Engineering moves at software speed. Manufacturing... ...Description: We are looking for a Machine Learning Engineer to join our team and help us... ...including robotic perception, control, or planning. *This job description is not...
Flexible hours
Machina Labs
Los Angeles, CA
3 days ago
Staff Machine Learning Engineer (Tapestry)
$197k - $288k
...Software Engineering Mountain View, CA (HQ) About the team: Tapestry... ...AI-enabled analytical and planning tools that allow the entire energy... ..., we would love to meet you. Learn more about our team and our... ...We're looking for a Staff Machine Learning Engineer to join our...
Full time
X Development, LLC
Mountain View, CA
1 day ago
Staff ML Engineer, Autonomy & Planning
$240k - $275k
...Company combines autonomous machines, advanced software, and human... ...for a Staff Autonomy Engineer to own thislayerend to end.... ...latest advances in embodied AI, learned planning, and real-time decision making... ...that drive safe, bounded robot behavior. ~ Build the closed-loop...
Full time
Knightscope
Sunnyvale, CA
4 days ago
Machine Learning Engineer- Remote
$110.8k - $166.1k
...Job Description The ML Engineer is responsible for the overall development, deployment, and support of our machine learning operations Harbor Freight. This includes the architecture... ...are able to enroll in our company's 401k plan. Associates will accrue paid time off up...
Temporary work
Work at office
Remote work
Flexible hours
Harbor Freight Tools
Calabasas, CA
2 days ago
Sr Machine Learning Engineer
$165k - $195k
...Senior Machine Learning Engineer The Marlin Alliance, Inc. is seeking a talented and experienced Senior Machine Learning Engineer to join... ...generation of governed, machine-consumable data. From strategic planning to advanced analytics and cybersecurity, our team provides...
Contract work
Work at office
The Marlin Alliance
San Diego, CA
4 days ago
Senior Machine Learning Engineer, MLOps West Coast
$131.4k - $235.95k
...creating software tools for making buildings, machines, and even the latest movies, we... ...the world. As a Senior Machine Learning Engineer focused on Machine Learning Ops (MLOps... ..., cost optimization, and capacity planning. Experience with observability tooling...
For contractors
Remote work
Autodesk
San Francisco, CA
1 day ago
Machine Learning Engineer
$1,000 per month
...Machine Learning Engineer Elicit is building the reasoning layer for science and decision-making. We use language models to search over 12... ...for target assessment, evidence synthesis, and experiment planning that allow models to provide guarantees about their processes...
Work at office
Remote work
Home office
Relocation package
Flexible hours
Elicit
Oakland, CA
5 days ago
Sr. Machine Learning Engineer
...organizations that keep the world running. Our Team's Vision: Our Engineering team is shaping the future of cybersecurity. We thrive on... ...Scale: Design sophisticated AI Agents capable of autonomous planning, memory management, and high-reliability tool-use across...
Immediate start
Illumio
Sunnyvale, CA
5 days ago
Senior ML Compiler Engineer
$128.7k - $261.3k
...perception, prediction, and planning research into production-... ...development, and performance engineering so that every cycle on our... ...reaction times, and more robust behavior on the road. If you want... ...developing and deploying machine learning models? Compensation:...
Local area
Work from home
Relocation package
Flexible hours
General Motors
San Francisco, CA
5 days ago
Senior Machine Learning Engineer
$184.5k
...world. Join us. Senior ML/Gen AI Engineer Introduction to the Team:... ...technology platform powered by data and machine learning provides secure, differentiated, and personalized... ...-making  Understanding of agent planning, memory, tool use, and self-reflection...
Local area
Flexible hours
Expedia Group
San Jose, CA
3 days ago
Machine Learning Engineer
...JOB SUMMARY The ML Engineer II will design and develop ML models that deliver accurate results to solve identified business problems... ...scientists on model deployment. • Work with product teams in planning and execution of new product releases. • Set OKRs and success...
Compunnel
Aliso Viejo, CA
3 days ago
Senior Machine Learning Engineer
$160k - $250k
...Senior Machine Learning Engineer In order to execute our vision, we need to grow our team of best-in-class machine learning engineers. We... ...ingenious implementations to the team, and is capable of planning out scalable, maintainable data pipelines. Responsibilities...
Hive
San Francisco, CA
5 days ago
Senior ML Accelerator Engineer - GPU
$128.7k - $261.3k
...perception, prediction, and planning research into production‑grade... ...development, and performance engineering so that every cycle on our accelerators... ...times, and more robust behavior on the road. If you want... ...on realizing your ambitions. Learn how GM supports a rewarding...
Local area
Work from home
Relocation package
Flexible hours
General Motors
Sunnyvale, CA
4 days ago
Senior Machine Learning Engineer, DevOps/SRE
$148.75k - $361k
...large scale and with low latency. We use Machine Learning, Reinforcement Learning, AI, Control... ...talented and experienced Senior Software Engineer, MLOps/DevOps, to join the Advertising... ...monitoring, drift detection, capacity planning, and pipeline health metrics Participate...
Work at office
Local area
Remote work
Monday to Thursday
Flexible hours
Roku
San Jose, CA
1 day ago
Senior Machine Learning Engineer - Fine-Tuning and On-device AI
$120k - $215k
...assembling a diverse, world-class team-engineers, designers, researchers, and product minds... ...the Role We are seeking a Senior Machine Learning Engineer to lead the fine-tuning,... ...edge applications such as orchestration, planning, multi-agent coordination, and other intelligent...
Full time
Temporary work
Local area
Flexible hours
HP IQ
Palo Alto, CA
5 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Machine Learning Engineering TL, Behavior Planning. Be the first to apply!