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

DiDi Labs

$169K — $351K *
Transportation
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • Master's or higher degree in a relevant technical field
  • 3-8+ years of experience in AI infrastructure and model optimization
  • Strong programming skills in C++ and Python
  • Expertise in parallel programming with CUDA and OpenMP
  • Familiarity with mainstream inference engines and LLM frameworks
  • Knowledge of GPU architectures and memory bandwidth management

Responsibilities

  • Optimize and schedule deployment of AI models for vehicles
  • Lead initiatives to ensure system stability and analyze performance issues
  • Architect service-oriented environments for Large Language Models
  • Integrate advanced optimization methods and monitor execution
  • Establish profiling and telemetry frameworks for hardware utilization
  • Collaborate with cross-functional teams for efficient algorithm deployment

Benefits

  • Comprehensive benefits package
  • Equity options and bonus structure
  • Supportive and innovative work culture
  • Opportunities for professional development and continuous learning
Full Job Description
About The Role

We are seeking an experienced and mission-driven Senior/Sr. Staff AI Infrastructure Engineer, Inference & Optimization to lead the performance tuning, deployment, and resource scheduling of cutting-edge AI models across on-vehicle and cloud infrastructure. In this role, you will design high-efficiency inference pipelines, build system-level stability frameworks, and optimize hardware execution to ensure ultra-low latency and rock-solid operational reliability. You will act as a technical leader in AI infrastructure, accelerating model iteration and bridging the gap between frontier deep learning algorithms and real-time autonomous systems.

Responsibilities
  • Own the deployment, optimization, and resource scheduling of vehicle-side AI models, ensuring high efficiency, low latency, and robust execution within embedded constraints.
  • Lead vehicle-side system stability initiatives, conducting independent root-cause analysis and driving resolution for complex, system-level performance bottlenecks and runtime anomalies.
  • Architect and scale service-oriented deployment environments for Large Language Models (LLMs) and foundational models to support offline simulation, automated annotation, and rapid model validation.
  • Track and evaluate cutting-edge industry methodologies, continuously integrating advanced optimization toolchains, quantization techniques, and execution engines.
  • Establish system-level profiling and telemetry frameworks using CUDA tools to monitor, analyze, and maximize hardware utilization across target GPU architectures.
  • Collaborate cross-functionally with Autonomous Driving Perception/Prediction, Cloud Infrastructure, and Safety teams to enable rapid algorithm iteration and scalable vehicle deployment.


Qualifications
  • Master's or higher degree in Computer Science, Software Engineering, Systems Engineering, or a closely related technical field.
  • 3-8+ years of industry experience in high-performance computing, AI infrastructure, model optimization, or embedded deployment.
  • Strong proficiency in C++ and Python, with solid expertise in parallel programming (CUDA, OpenMP) and low-level system profiling tools.
  • Deep familiarity with mainstream inference engines (e.g., TensorRT, ONNX Runtime) and specialized LLM inference/serving frameworks (e.g., vLLM, SGLang, TensorRT-LLM).
  • Practical understanding of modern GPU hardware architectures (e.g., NVIDIA Hopper, Thor) and memory bandwidth management.
  • Demonstrated ability to diagnose complex software-hardware integration issues and drive scalable, production-grade solutions.

Preferred Qualifications
  • Hands-on experience optimizing and deploying AI models on the NVIDIA Thor platform, including hardware resource scheduling and acceleration.
  • Proven track record of serving large foundation models (e.g., LLaMA, Qwen, GPT) in production or high-throughput cloud pipelines using frameworks like vLLM, SGLang, TGI, or LightLLM.
  • Background in deep learning training frameworks (PyTorch) and practical experience with model quantization (INT8/FP8/AWQ), kernel fusion, or graph compilation.
  • Experience deploying real-time, high-availability AI workloads in autonomous vehicles, robotics, or edge devices.


The base salary range for this full-time position is $169,783 - $351,000 annually in addition to bonus, equity and benefits. Our salary ranges are determined by role, level, and location. Within the range, individual pay is determined by work location and additional factors, including job-related skills, experience, and relevant education or training.

Similar Jobs

More Jobs at DiDi Labs

More Transportation Jobs

Find similar Senior/Sr. Staff AI Infrastructure Engineer, Inference & Optimization jobs: