Machine Learning Engineer - Foundation Model

Job Description

XPENG, a smart technology company focused on AI and autonomous driving technologies, is seeking a Machine Learning Engineer. The engineer will contribute to building a state-of-the-art ML infrastructure for training large foundation models and accelerating model training/inference. The ideal candidate will work with a team of talented engineers and scientists to advance machine learning models for autonomous driving solutions.

Responsibilities:

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

Requirements:

• Master's degree in CS/CE/EE or equivalent.
• 1-3 years of industry experience.
• Good knowledge of PyTorch.
• Knowledge of model training framework (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 the work.
• Opportunity to work on cutting-edge technologies with top talent.
• Opportunity to impact transportation through autonomous driving.
• Competitive compensation package.
• Snacks, lunches, dinners, and fun activities