Machine Learning Engineer Intern - Foundation Models

XPeng Motors is one of China’s leading smart electric vehicle (EV) companies. We design, develop, and manufacture smart EVs that are seamlessly integrated with advanced Internet, AI and autonomous driving technologies. We are committed to in-house R&D and intelligent manufacturing to create a better mobility experience for our customers. We strive to transform smart electric vehicles with technology and data, shaping the mobility experience of the future.

We are looking for machine learning engineers with strong research and development skills, system design skills and software development skills. You will design and implement foundation models for different downstream tasks for our autonomous driving system. You will closely work with experts from perception, prediction, planning teams to make our vehicles drive intelligently on various road conditions.

You will be working with a team of the best-in-class computer vision, AI systems, and software engineers to ensure the world-leading performance on our autonomous vehicles. Your work will be supported by massive data from our autonomous fleet to deliver the best autonomous driving solution.

Job Responsibilities:

• Research and develop foundation models for multiple perception tasks, e.g., object detection, segmentation, etc., to achieve the SOTA performance.

• Develop and integrate software based on recent self-supervised learning techniques and pre-training strategies.

• Build data pipelines for better foundation model pre-training.

• Deliver pre-trained and finetuned models to other perception team to achieve the most intelligent autonomous driver in class.

• Work with massive field-testing data to continuously improve autonomous driving technologies.

Minimum Skill Requirements:

• PhD or Master in computer science, electrical engineering, or other related fields.

• Strong experience in applied machine learning including data collection, data analysis, feature engineering and model training.

• Solid understanding on data structures, algorithms, code optimization and large-scale data processing.

• Proficient in developing algorithms in Python or C++.

• Familiar with common Deep Learning frameworks, e.g., PyTorch.

• Excellent problem-solving skills.

Preferred Skill Requirements:

• Experience in Computer Vision, Machine Learning, Robotics, Natural Language Processing, or similar technical fields.

• Experience in foundation model or large language model development.

• Experience in CNN/RNN/GNN, imitation learning and reinforcement learning.

• Experience in GPU/CUDA/TensorRT.

What do we provide:

• A fun, supportive and engaging environment

• Opportunity to make significant impact on the transportation revolution by the means of advancing autonomous driving

• Opportunity to work on cutting edge technologies with the top talent in the field

• Competitive compensation package

• Snacks, lunches and fun activities

We are an Equal Opportunity Employer. It is our policy to provide equal employment opportunities to all qualified persons without regard to race, age, color, sex, sexual orientation, religion, national origin, disability, veteran status or marital status or any other prescribed category set forth in federal or state regulations.