Staff Machine Learning Engineer - Foundation Model

Job Description

XPENG, an innovative smart technology company, is seeking a Staff Machine Learning Engineer to join their team in Santa Clara, CA. The ideal candidate will be passionate about creating a state-of-the-art ML infrastructure for training large foundation models and accelerating model training/inference. This role involves working alongside talented engineers and scientists to advance machine-learning models for autonomous driving.

Responsibilities:

• Design, train, and deploy large deep learning models.
• Leverage vast amounts of labeled and unlabeled data from a fleet of vehicles.

Requirements:

• Master's degree in CS/CE/EE or equivalent with 5+ years of industry experience.
• Strong knowledge of PyTorch.
• Knowledge of model training frameworks (e.g., PyTorch Lightning).
• Knowledge of transformer architectures.
• Experience with PyTorch DDP for distributed training.

What XPENG offers:

• A fun, supportive, and engaging environment.
• Infrastructures and computational resources.
• Opportunity to work on cutting-edge technologies.
• Opportunity to impact the transportation revolution.
• Competitive compensation package.
• Snacks, lunches, dinners, and fun activities.