XPENG Motors
Senior Machine Learning Engineer - AI Foundation
$175K–$296K
Santa Clara, CA +1 · Senior · Full-time · Posted Sep 16, 2026
- Location
- Santa Clara, CA
Mountain View, California, United States - Experience
- Experience performing large-scale distributed training of models; no years specified.
- Education
- Master's degree in CS/CE/EE, or equivalent industry experience.
About the job
Build ML infrastructure and accelerate training and inference for large physical AI foundation models supporting autonomous driving and related applications.
What you'll do
- Design data pipelines streaming labeled and unlabeled fleet data at scale.
- Implement training frameworks for physical AI foundation models, including VLA 2.0, XWorld, and Robotics.
- Improve distributed model training and cloud inference speed for simulation, reinforcement learning, and enterprise LLM/VLM applications.
What they ask for
- Master's degree in CS/CE/EE, or equivalent industry experience.
- Deep knowledge of PyTorch and transformer architecture.
- Knowledge of model inference frameworks such as vLLM or SGLang.
- Experience with large-scale distributed model training and profiling to improve training and inference speed.
Nice to have
- Autonomous driving industry experience.
- CUDA custom ops, edge computing, or distributed computing frameworks such as Ray.
Pay
$174,720-$295,680 USD base salary
Tools
PyTorch, vLLM, SGLang, FSDP, Expert Parallel, Context Parallel, CUDA, Ray
AI focus
Develop training and inference infrastructure for physical AI foundation models and accelerate models used in autonomous driving, closed-loop simulation, reinforcement learning, and enterprise LLM/VLM applications.
Model-written summary · the original posting is authoritative