XPENG is a leading smart technology company at the forefront of innovation, integrating advanced AI and autonomous driving technologies into its vehicles, including electric vehicles (EVs), electric vertical take-off and landing (eVTOL) aircraft, and robotics. With a strong focus on intelligent mobility, XPENG is dedicated to reshaping the future of transportation through cutting-edge R&D in AI, machine learning, and smart connectivity.
We are looking for a full-time Machine Learning Engineer, with deep knowledge and strong enthusiasm towards establishing a state-of-art ML infrastructure for training very large foundation model and accelerating model training/inference.
Our mission is to solve the autonomous driving problem. You will work with a team of talented software engineers, machine learning engineers and research scientists to push the boundary of state-of-art machine learning models which will enable the next-generation E2E solution of autonomous driving.

MINIMUM SKILL REQUIREMENTS:

• Master in CS/CE/EE, or equivalent, with 1-3 years of industry experience.
• Good knowledge of PyTorch.
• Knowledge of model training framework (e.g. PyTorch Lightning)
• Knowledge of transformer architecture and ways to accelerate the training and inference of transformer models.
• Experience of using pytorch ddp for distributed training of models.

PREFERRED SKILL REQUIREMENTS:

• Strong publications records in top academic conferences or journals, e.g. CVPR, NeurIPS, ICML, ICCV, ECCV, ICLR
• Experience in training large scale vision or language models
• Previous experience in the autonomous driving industry.
• Being efficiently in solving complex problems collaboratively on larger teams

• Design, train, and deploy large deep learning models that can leverage the vast amount of labeled and unlabeled data from a fleet of million vehicles.