XPENG · 2 months ago
Staff Machine Learning Engineer - AI Foundation
Santa Clara, CA
Full-time
Onsite
Lead/Staff
$215K/yr - $364K/yrXPENG is a leading smart technology company at the forefront of innovation, integrating advanced AI and autonomous driving technologies into its vehicles. They are seeking a full-time Staff Machine Learning Engineer - AI Foundation to establish a state-of-art ML infrastructure for training large foundation models and accelerating model training/inference.
AutomotiveAutonomous VehiclesElectric VehicleManufacturing
Responsibilities
Design, train, and deploy large deep learning models that can leverage the vast amount of labeled and unlabeled data from a fleet of million vehicles
Qualification
Required
PhD in CS/CE/EE, or equivalent, in industry experience
Deep knowledge of PyTorch
Knowledge of model training framework (e.g. PyTorch Lightning)
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
Preferred
Previous experience in the autonomous driving industry
Experience with cuda or Triton language for writing custom ops
Knowledge of Nvidia TensorRT
Experience with edge computing systems
A track record of efficiently solving complex problems collaboratively on larger teams
Benefits
Competitive compensation package
Snacks, lunches, dinners, and fun activities
Company
XPENG
XPeng is a leading Chinese Smart EV company that designs, develops, manufactures, and markets Smart EVs that appeal to the large and growing base of technology-savvy middle-class consumers.
Founded in 2014
10001+ employees
Funding
Current Stage
Public CompanyTotal Funding
$7.8BKey Investors
China CITIC BankVolkswagen GroupAgricultural Bank of China
2025-08-18Post Ipo Debt· $1.39B
2023-07-26Post Ipo Equity· $700M
2022-04-27Post Ipo Debt· $1.14B
Leadership Team
Heng Xia
Co-Founder/ Presient
Xiaopeng He
Chairman, Co-founder
Recent News
Company data provided by crunchbase