小鹏汽车【26届校招】AI编译工程师

任职要求

【任职要求】
1.本科及以上, 有良好的计算机理论基础，熟悉体系结构者优先，包括GPU，CPU，NPU，DSP等
2.优秀的python和C/C++编程能力，熟悉cuda以及pytorch等优先
3.有AI模型训练和推理经验优先 
4.有编译器开发经验优先

工作职责

主要负责模型量化和编译器工具的开发，工作内容包括：
1. 量化工具的开发，包括模型平滑，伪量化，定点化等
2. 算子计算优化，以及算子融合和图优化等
3. 并行以及切分策略等
4. 算子代码生成codegen
5. 编译器后端，包括控制流优化，指令调度，寄存器和内存管理，物理代码生成等