Senior Staff AI Performance Engineer

Job Description

XPENG, a smart technology company focused on AI and autonomous driving technologies, is seeking a Senior Staff AI Performance Engineer. The ideal candidate will optimize the training and inference performance of machine learning infrastructure and foundation models for autonomous driving.Role responsibilities include:

• Optimizing the performance of Deep Learning training workload on NVIDIA GPU systems on a large scale.
• Optimizing the latency of model inference and model pre- and post-processing on AV onboard systems.

Minimum requirements include:

• PhD in CS/CE/EE, or equivalent industry experience.
• 4+ years of C++ and Python programming experience.
• Knowledge of Computer Architecture and Operating Systems.
• Background with parallel programming, preferably on GPUs.
• Ability to analyze/profile workloads and reduce system latency.
• A proven track record of delivering CPU or GPU latency optimization for real-world engineering systems.

XPENG offers:

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