DiDi Labs

MachineLearningEngineer,VLA

$129–247k Mountain View, California, United States
Market Sentiment
HIGH DEMAND

Neural analysis suggests this role is
optimal for Mid+ candidates.

The Brief

“Machine Learning Engineer, VLA at DiDi Labs. Skills: Machine learning, Deep learning, Autonomous driving, Foundation models. Design deep learning models. Train deep learning models”

What They're Looking For.

Must Have

background in machine learning, background in deep learning, background in robotics, Experience with PyTorch, Experience with JAX, Experience with TensorFlow, Solid understanding of modern neural architectures, coding skills in Python, coding skills in C++

Nice to Have

BS in Computer Science, MS in Computer Science, PhD in Computer Science, BS in Machine Learning, MS in Machine Learning, PhD in Machine Learning, BS in Robotics, MS in Robotics, PhD in Robotics, Experience in autonomous driving, Experience in robotics, Experience in embodied AI, Experience training large models, Experience with trajectory prediction, Experience with planning, Experience with decision-making systems, Publications in top ML conferences, Publications in top robotics conferences

What You'll Do.

Design deep learning models

Train deep learning models

Build foundation model architectures

Develop scalable training pipelines

Work with massive datasets

Build high-quality data pipelines

Optimize models for latency

Optimize models for reliability

Optimize models for on-vehicle deployment

Collaborate with perception teams

Collaborate with mapping teams

Collaborate with simulation teams

Collaborate with systems teams

Drive research ideas into production

How You'll Work.

Team & Collaboration

perception teams; mapping teams; simulation teams; systems teams

Full Job Description

About The Company DiDi's autonomous driving unit was established in 2016 with the mission of developing Level 4 autonomous driving (AD) technology to make transportation safer and more efficient. In August 2019, the unit became an independent company, DiDi Autonomous Driving, dedicated to advanced AD R&D, product application, and business expansion. We believe integrating AD technology into a shared-mobility fleet will generate immense social value. By leveraging DiDi's specialized technology, operational expertise, and integrated ecosystem, we are positioned to build and operate a highly efficient, user-oriented autonomous fleet. About The Role The VLA Team focuses on building large-scale foundation models for multi-agent behavior prediction and autonomous vehicle planning. By leveraging DiDi Voyager’s unparalleled driving data, we train highly robust and generalizable deep learning systems that enable safe and intelligent autonomous driving at scale. Our models serve as the core intelligence of the autonomous driving stack, enabling vehicles to understand complex traffic scenarios, anticipate agent behavior, and make safe and efficient driving decisions. We operate at the intersection of large-scale machine learning, autonomous driving, and foundation model research, pushing the frontier of multi-agent prediction and planning. Responsibilities As a member of the VLA Team, you will: Design and train large-scale deep learning models for: Multi-agent trajectory prediction Behavior and intent prediction Planning and decision-making Build foundation model architectures (Transformers, Diffusion, Flow-based models, Decision models, VLM/VLA) Develop scalable training pipelines across hundreds to thousands of GPUs Work with massive real-world datasets and build high-quality data pipelines Optimize models for latency, reliability, and on-vehicle deployment Collaborate closely with perception, mapping, simulation, and systems teams Drive research ideas into production system

Free ATS check

Applying for this Machine Learning Engineer, VLA role?

Most applicants get filtered before a human reads their resume. See if yours makes the cut.

How to Apply on Greenhouse

  • Create a Greenhouse profile before applying — it saves time across multiple applications.
  • Upload your resume as a PDF; the parser handles it better than Word.
  • Answer all knockout questions carefully — wrong answers auto-reject before a human sees you.
  • Enable email notifications to track application status in real time.

ANONYMOUS · UNFILTERED

What do employees actually say about DiDi Labs?

Real rants from real employees. Read before you apply.

Read Company Rants →