Senior Staff Machine Learning Engineer - Foundation Model

Job Description

XPENG, a smart technology company focused on AI and autonomous driving, is seeking a Senior Staff Machine Learning Engineer to join their team in Santa Clara, CA. This role involves establishing a state-of-the-art ML infrastructure for training large foundation models and accelerating model training/inference. The engineer will collaborate with talented software engineers, machine learning engineers, and research scientists to advance autonomous driving technology.

Responsibilities include:

• Designing, training, and deploying large deep learning models using data from a fleet of vehicles.

Requirements include:

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

The role offers:

• 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 significant impact on the transportation revolution by the means of advancing autonomous driving.
• Competitive compensation package.
• Snacks, lunches, dinners, and fun activities.