DiDi · 5 hours ago
Machine Learning Engineer, Foundation Model
San Jose, CA
Full-time
Onsite
Mid, Senior Level
$129K/yr - $247K/yrDiDi is focused on developing Level 4 autonomous driving technology aimed at enhancing transportation safety and efficiency. The role involves building large-scale foundation models for multi-agent behavior prediction and autonomous vehicle planning, leveraging extensive driving data to enable intelligent decision-making for autonomous vehicles.
Artificial Intelligence (AI)Mobile AppsPublic TransportationRide SharingTransportation
Responsibilities
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 systems used by real autonomous vehicles
Qualification
Required
Strong background in machine learning, deep learning, or robotics
Experience with PyTorch / JAX / TensorFlow
Solid understanding of modern neural architectures (transformers, diffusion, auto-regressive)
Strong coding skills in Python and C++
Passion for building real-world, safety-critical AI systems
Preferred
BS, MS or PhD in Computer Science, Machine Learning, Robotics, or a related field
Experience in autonomous driving, robotics, or embodied AI
Experience training large models on distributed GPU clusters
Experience with trajectory prediction, planning, or decision-making systems
Publications in top ML / robotics conferences (NeurIPS, ICML, ICLR, CVPR, RSS, CoRL, etc.)
Benefits
Bonus
Equity
Benefits
Company
DiDi
Didi is a mobility technology platform that offers ride hailing, taxi hailing, food delivery, and financial services.
Founded in 2012
10001+ employees
H1B Sponsorship
DiDi 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 (2)
2024 (4)
2023 (2)
2022 (16)
2021 (30)
2020 (12)
Funding
Current Stage
Public CompanyTotal Funding
$23.4BKey Investors
Toyota MotorBooking HoldingsMirae Asset
2023-10-12Post Ipo Equity· $149M
2021-06-30IPO
2021-04-09Debt Financing· $1.5B
Leadership Team
Cat Dumitru
Global Performance Marketing & Partnerships
Recent News
New Enterprise Associates
2025-12-29
2025-12-03
Company data provided by crunchbase