Staff Machine Learning Engineer - AI Foundation

Job Description

XPENG, a smart technology company focused on AI and autonomous driving, is seeking a Staff Machine Learning Engineer - AI Foundation. This role involves establishing a state-of-the-art ML infrastructure to accelerate model training and inference. The ideal candidate will collaborate with software engineers, machine learning engineers, and research scientists to advance autonomous driving solutions.

Responsibilities:

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

Requirements:

• PhD in CS/CE/EE or equivalent industry experience.
• Deep knowledge of PyTorch.
• Knowledge of model training frameworks (e.g., PyTorch Lightning).
• In-depth knowledge of transformer architecture.
• Experience in large-scale distributed training of models.
• Experience profiling models and improving training/inference speed.

What XPENG Offers:

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