Senior / Staff Machine Learning Engineer, Behavior Planning & Prediction

Woven by Toyota is enabling Toyota’s once-in-a-century transformation into a mobility company. Inspired by a legacy of innovating for the benefit of others, our mission is to challenge the current state of mobility through human-centric innovation — expanding what “mobility” means and how it serves society.

Our work centers on four pillars: AD/ADAS, our autonomous driving and advanced driver assist technologies; Arene, our software development platform for software-defined vehicles; Woven City, a test course for mobility; and Cloud & AI, the digital infrastructure powering our collaborative foundation. Business-critical functions empower these teams to execute, and together, we’re working toward one bold goal: a world with zero accidents and enhanced well-being for all.

TEAM

The Behavior team at Woven by Toyota tackles autonomy challenges in prediction and trajectory planning. Our work involves a variety of technical hurdles, such as analyzing petabytes of multimodal driving data, solving optimization problems, minimizing latency on hardware accelerators, deploying scalable and efficient machine learning (ML) training and evaluation pipelines, and designing novel neural network architectures to advance state-of-the-art ML for prediction and motion planning. We are looking for driven and creative problem-solvers to join us in improving mobility for everyone through human-centered automated driving solutions for personal and commercial applications.

WHO ARE WE LOOKING FOR?

The Behavior team is searching for a skilled Machine Learning Engineer to support the entire lifecycle of machine learning models. This role involves developing and optimizing machine learning systems, including performance tuning, model development, and developing infrastructure and support systems that facilitate this work. Responsibilities include developing data pipelines, with a focus on transform libraries and dataset management, contributing to metrics and metrics pipelines, and integrating models into our on-vehicle software. This is an opportunity to design and implement innovative machine learning models for our next-generation autonomous vehicle platform, which will impact millions of Toyota production vehicles. We are seeking individuals who are passionate about self-driving car technology and its transformative potential for humanity.

Responsibilities

• Design and develop advanced machine learning models in the behavior space, specifically tailored for autonomous vehicles, utilizing deep learning and large-scale data analysis.
• Deploy scalable and efficient ML models on our autonomous vehicle platform.
• Integrate modern technologies with rigorous safety standards while maintaining cost efficiency.
• Oversee the development of new ML models end-to-end, from data strategy and initial development to optimization, production platform validation, and fine-tuning based on metrics and on-road performance.
• Design, build, and maintain ML pipelines that produce training and evaluation data for behavior and prediction models, including ETL, feature extraction, augmentation, and labeling workflows.
• Implement auxiliary pipelines for model validation, metric computation, and on-road performance integration.
• Lead large, multi-person projects and significantly influence the overall motion planning architecture and technical direction.
• Enable and support other engineers by coaching, leading by example, and providing high-quality code and design document reviews, as well as delivering rigorous reports from ML experiments.
• Contribute significantly to the development of essential components for end-to-end ML training and deployment, from data strategy to optimization and validation.
• Be a champion of the scientific method and critical thinking to invent state-of-the-art deep learning solutions.
• Work in a high-velocity environment and employ agile development practices.
• Collaborate closely with teams such as Perception, Simulation, Infrastructure, and Tooling to drive unified solutions.

Minimum Qualifications

• MS or PhD in Machine Learning, Computer Science, Robotics, or related quantitative fields, or equivalent industry experience.
• 3+ years of experience with Python, major deep learning frameworks, and software engineering best practices.
• Expertise with deep learning approaches, such as supervised/unsupervised learning, transfer learning, multi-task learning, and deep reinforcement learning.
• Built robust data pipelines, ETL workflows, and integration systems to support behavior prediction models
• Good understanding of learning-based planning approaches, including imitation learning, reinforcement learning, and state-of-the-art techniques for sequential modeling, such as Transformer architectures.
• Practical experience with ML training and evaluation workflows, including data sampling, preprocessing, and reproducible experiment management.
• 3+ years of experience covering machine learning workflows, data sampling and curation, preprocessing, model training, ablation studies, evaluation, deployment, and inference optimization.
• Comfortable writing C++ code for integration with our autonomous vehicle platform.
• Passion for self-driving car technology and its potential to impact humanity.
• Strong communication skills with the ability to articulate concepts clearly and precisely.

NICE TO HAVES

• Published research at top-tier conferences (NeurIPS, RSS, IROS, ICRA, etc.).
• Proven track record of deploying ML models at scale in self-driving or related fields.
• Familiarity with production-level coding in time-limited task schedules.
• Experience with robot motion planning (e.g., trajectory optimization, sampling-based planning, model predictive control).
• Experience working with temporal data and sequential modeling.
• Expertise in self-driving challenges (Perception, Prediction, Planning, Simulation).

For positions based in Palo Alto, CA, the base pay ranges from $161000- $264500 a year.

Your base salary is one part of your total compensation. We offer a base salary, short term and long term incentives, and a comprehensive benefits package. The total compensation offered to an employee will be dependent upon the individual's skills, experience, qualifications, location, and level.

What We Offer

We are committed to creating a modern work environment that supports our employees and their loved ones. We offer many options of the best programs to allow you to do your most meaningful work and to help you shape the future of mobility.

・Excellent health, wellness, dental and vision coverage

・A rewarding 401k program

・Flexible vacation policy

・Family planning and care benefits

Our Commitment

・We are an equal opportunity employer and value diversity.

・Any information we receive from you will be used only in the hiring and onboarding process. Please see our privacy notice for more details.