Senior Machine Learning Engineer

Home > Python programming jobs

Senior Machine Learning Engineer in USA

XPENG
Full time
Email

USA
09/18/2026
Report

Santa Clara, CA

Responsibilities

Design and implement training data pipelines streaming labeled and unlabeled data at hundreds-of-petabytes scale from more than one million vehicles.
Implement training frameworks for XPENG physical AI foundation models, including VLA 2.0, XWorld, and Robotics.
Accelerate model training using FSDP, Expert Parallel, Context Parallel, and advanced data types.
Accelerate cloud model inference for closed-loop simulation, reinforcement learning, and enterprise LLM/VLM applications.
Profile models and investigate performance bottlenecks to improve training and inference speed.

Requirements

Master's degree in computer science, computer engineering, or electrical engineering, or equivalent industry experience.
Deep knowledge of PyTorch.
Knowledge of model inference frameworks such as vLLM or SGLang.
In-depth knowledge of transformer architecture and techniques for accelerating transformer training and inference.
Experience with large-scale distributed model training.
Preferred experience in the autonomous driving industry.
Preferred experience writing custom operations with CUDA.
Preferred experience with edge computing systems.
Preferred knowledge of distributed computing frameworks such as Ray.
Ability to solve complex problems collaboratively on larger teams.

Benefits

Full-time position with a supportive and engaging work environment.
Access to infrastructure and computational resources to support the work.
Opportunity to work with leading talent on cutting-edge technology and impact autonomous driving.
Competitive compensation package with bonus, equity, and benefits.
Snacks, lunches, dinners, and fun activities.

Salary: $175k - $296k/yr

XPENG

XPENG designs, manufactures, and sells smart electric vehicles for consumers, integrating in-house advanced driver-assistance systems and connected in-car software. Founded in 2014 and headquartered in Guangzhou, it operates plants in Zhaoqing and Guangzhou, maintains a European HQ in Amsterdam, ...

Similar positions

Senior, Machine Learning Engineer

Torc Robotics
Full time

USA
09/22/2026
Salary: $177k - $213k/yr

Remote

Senior Software Engineer

Commonwealth Bank of Australia
Full time

Australia
09/22/2026

Sydney

Deployed Engineer, Professional Services

LangChain
Full time

Remote
09/22/2026
Salary: Competitive

APAC

Senior AI Engineer

Jeeves
Full time

Remote
09/22/2026

Brazil

Senior Product Security Engineer

AlphaSense
Full time

USA
09/22/2026
Salary: $182k - $228k/yr

Remote