Drimva

DiDi Global

Senior/Sr. Staff AI Infrastructure Engineer, Inference & Optimization

$170K–$351K

San Jose, CA +1 · Senior Staff · Full-time · Posted Sep 11, 2026

Location
San Jose, CA
Mountain View, California, United States
Experience
3-8+ years of industry experience.
Education
Master’s or higher degree in Computer Science, Software Engineering, Systems Engineering, or a closely related technical field.

About the job

Leads deployment, performance optimization, and resource scheduling for AI models across autonomous vehicles and cloud infrastructure, supporting real-time systems and model validation.

What you'll do

  • Deploy and optimize vehicle-side AI models, including resource scheduling within embedded constraints.
  • Lead system stability, profiling, and root-cause analysis to improve performance and reliability.
  • Build scalable serving environments for LLMs and foundational models used in simulation, annotation, and model validation.

What they ask for

  • Master’s or higher degree in a related technical field.
  • 3-8+ years of industry experience in high-performance computing, AI infrastructure, model optimization, or embedded deployment.
  • Strong C++ and Python skills, parallel programming experience with CUDA and OpenMP, and low-level profiling expertise.
  • Experience with inference engines and LLM serving frameworks, plus modern GPU architectures and memory bandwidth management.

Nice to have

  • Experience optimizing and deploying models on NVIDIA Thor.
  • Production experience serving foundation models; experience with PyTorch, quantization, kernel fusion, graph compilation, or real-time AI workloads in vehicles, robotics, or edge devices.

Pay

$169,783 - $351,000 annually (USD), base salary

Tools

C++, Python, CUDA, OpenMP, TensorRT, ONNX Runtime, vLLM, SGLang

AI focus

AI infrastructure for autonomous driving, including vehicle-side inference optimization and cloud-based LLM and foundational-model serving.

Model-written summary · the original posting is authoritative

First seen Oct 29, 2026 · Last verified Nov 5, 2026 (UTC)