Career EngineCareer Engine
Your matchExample
83%match
Why you fitThis role calls for Llm and Pytorch — tell us what you know and we'll check the rest.
Gap to noteAlso lists Ray — worth confirming before you apply.

Get roles like this, scored and matched, in your inbox.

Sign up free →

Job Opening

Staff/Principal AI Transformation Engineer

DiDi Labs · San Jose, CA · Posted 18 days ago

$255k–$351kmgr
8+ yrs expLLMPyTorchRayvLLMTriton Inference ServerKubernetesDeepSpeedMegatron-LM

Own the architecture of a unified distributed AI platform and embed with Perception, Prediction, Planning, and Simulation teams to drive DiDi's company-wide shift to an AI-Native R&D organization.

At DiDi Autonomous Driving, we firmly believe that the future of mobility goes beyond simply "utilizing AI"—it will be fundamentally reimagined and entirely driven by an AI-Native architecture .

We are seeking a visionary, highly technical, and mission-driven Staff or Principal AI Transformation Engineer to act as the ultimate catalyst for our company-wide AI transformation. In this strategic, high-impact leadership role, you will combine cutting-edge Large Language Model (LLM) expertise, robust systems architecture design, and a proven track record of enterprise-level AI scaling. You will embed deeply with our core engineering teams to evolve our traditional R&D organization into a truly AI-Native powerhouse.

Key Responsibilities

  • AI Infrastructure & Platform Architecture: Spearhead the evaluation, selection, and deep integration of frontier LLM ecosystems (e.g., Llama, Hugging Face) and commercial AI platforms. Own the architectural design of our unified, distributed AI platform spanning complex data processing, model training, inference pipelines, and evaluation frameworks.
  • Forward Deployed Execution: Embed directly with core autonomous driving teams (Perception, Prediction, Planning & Control, and Simulation) via the FDE model. Pinpoint engineering bottlenecks, eliminate friction, and translate complex AI capabilities into production-ready internal ecosystems (e.g., AI DevOps, AI Copilots).
  • LLMOps / MLOps Orchestration & Optimization: Design and implement highly resilient, scalable automation pipelines for LLM deployment, monitoring, and continuous feedback loops. Optimize GPU cluster utilization, minimize inference latency, and maximize throughput across large-scale production environments.
  • Technical Roadmap & Vision: Keep a strong pulse on breakthrough trends in AGI and systems engineering. Act as a "super-connector" between external technological innovations and internal systems, ensuring our AI infrastructure maintains a 1-3 year competitive edge.

…

Apply on DiDi Labs's site →

This posting is sourced from DiDi Labs's own career site — applying takes you there directly.

Get roles like this, scored and matched, in your inbox.

Sign up free →

Related roles

There are 11 open roles at DiDi Labs right now. This role centers on Llm, Pytorch, Ray, and Vllm.

Want roles like this found for you?

Set your Focus once. Career Engine screens every new opening every week.

A scored, explained shortlist — not a keyword match — delivered weekly. Free for one profile.