Senior Staff Machine Learning Compiler Engineer

Job Description

XPENG, a smart technology company focused on AI and autonomous driving, is seeking a Senior Staff Machine Learning Compiler Engineer to develop a state-of-the-art machine learning compiler infrastructure for deploying advanced self-driving models on general-purpose GPUs. This role involves working within a talented team of software engineers, machine learning engineers, and research scientists to advance autonomous driving solutions.Responsibilities include:

• Developing machine learning compilers for GPU execution.
• Developing Clang/MLIR/LLVM compiler tool chain, including c/c++, assembler, linker.
• Implementing frontend optimizations, including computational graph optimization.
• Implementing backend optimizations via heterogeneous memory management, tiling, and parallelization.
• Testing and profiling compiler-deployed models.

Requirements include:

• A PhD in CS/CE/EE, or equivalent, with 5-7+ years of industry experience.
• Excellent C/C++ development capability.
• Previous experience in the autonomous driving industry(Preferred).
• Experience using Multi-Level IR Compiler Framework(Preferred).
• Familiarity with CUDA and PyTorch(Preferred).
• Familiarity with GPU CPU, NPU, DSP architecture(Preferred).
• A track record of efficiently solving complex problems collaboratively on larger teams(Preferred).

XPENG offers:

• A fun, supportive, and engaging environment.
• Infrastructures and computational resources to support your work.
• Opportunity to work on cutting-edge technologies with top talents.
• Opportunity to make a significant impact on the transportation revolution.
• Competitive compensation package.
• Snacks, lunches, dinners, and fun activities.