Posted by the employer · Santa Clara, United States
Senior Machine Learning Engineer - AI Foundation
XPENG · Santa Clara · United States
Posted: 9/17/2026 · On the employer's job feed since 10/11/2026
The advert
Quoted in full from XPENG's job posting. Gavroche does not edit it.
-
Design and implement training data pipeline that streams data from hundreds of petabytes of labeled and unlabeled data from a fleet of over a million vehicles.
-
Implement training framework for all physical AI foundation models in XPeng, including VLA 2.0, XWorld, Robotics.
-
Accelerate training with state of the art parallelisms, e.g., FSDP, Expert Parallel, Context Parallel, and data types.
-
Accelerate model inference on the cloud for closed-loop simulation, reinforcement learning, and enterprise LLM/VLM applications.
-
Master's Degree in CS/CE/EE, or equivalent, in industry experience.
-
Deep knowledge of PyTorch.
-
Knowledge of model inference framework (e.g. vLLM, SGLang)
-
In-depth knowledge of transformer architecture and ways to accelerate the training and inference of transformer models.
-
Experience of performing large scale distributed training of models.
-
A track record of profiling model and doing detective work to improve model training and inference speed.
-
Previous experience in the autonomous driving industry.
-
Experience with CUDA language for writing custom ops.
-
Experience with edge computing systems.
-
Knowledge of disributed computing frameworks, such as Ray.
-
A track record of efficiently solving complex problems collaboratively on larger teams
-
A fun, supportive and engaging environment.
-
Infrastructures and computational resources to support your work.
-
Opportunity to work on cutting edge technologies with the top talents in the field.
-
Opportunity to make a significant impact on the transportation revolution by the means of advancing autonomous driving.
-
Competitive compensation package.
-
Snacks, lunches, dinners, and fun activities.
Similar roles in Santa Clara
- AI Engineer · Teserac, Inc., Santa Clara
- Staff Machine Learning Engineer · XPENG, Santa Clara
- Staff Machine Learning Engineer - LLM Quantization & Deployment · XPENG, Santa Clara
- Staff Machine Learning Engineer - Foundation Model · XPENG, Santa Clara
- Machine Learning Engineer · Gatik AI, Santa Clara
- Machine Learning Engineer · Voltai, Palo Alto
200 open data scientist roles found; at most three per employer shown. Source: Direct employer job feeds (ATS) and career sites crawled by gavroche-direct.
How data scientist roles in United States are worked
8% fully remote · 2% hybrid · 90% on-site
From 91 adverts whose ATS states a work mode. Source: Direct employer job feeds (ATS) and career sites crawled by gavroche-direct; the employers' own work-mode flags (adverts without one count as on-site).
New roles in Santa Clara, last 12 weeks
- 7/20/2026 14
- 7/27/2026 10
- 8/3/2026 14
- 8/10/2026 17
- 8/17/2026 15
- 8/24/2026 22
- 8/31/2026 40
- 9/7/2026 15
- 9/14/2026 26
- 9/21/2026 34
- 9/28/2026 42
- 10/5/2026 39
288 vacancies. Source: Direct employer job feeds (ATS) and career sites crawled by gavroche-direct; vacancies by their own publication week.
Other roles at XPENG
- Senior Software Engineer - Agentic Platforms · Santa Clara
- Senior Machine Learning Engineer - Foundation Model · Santa Clara
- Senior Machine Learning Engineer - Predictive World Model · Santa Clara
- Senior Staff Machine Learning Engineer - Autonomous Driving Foundation Models · Santa Clara
- Staff Machine Learning Engineer · Santa Clara
- Staff Machine Learning Engineer - LLM Quantization & Deployment · Santa Clara
25 open. Source: Direct employer job feeds (ATS) and career sites crawled by gavroche-direct.