Machine Learning Engineer, Foundation Model

DiDi

27 days ago

San Jose, CA, USA

Mid Level / Senior

Base Salary

$129k - $247k/yr

Responsibilities

Design and train large-scale deep learning models for multi-agent trajectory prediction, behavior and intent prediction, and planning.
Build foundation model architectures including Transformers and Decision models.
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.

Strong background in machine learning, deep learning, or robotics.
Experience with PyTorch, JAX, or TensorFlow.
Solid understanding of modern neural architectures such as transformers and diffusion models.
Strong coding skills in Python and C++.
Passion for building real-world, safety-critical AI systems.

C++PythonPyTorchTensorFlow