NPU算法分析与算子优化工程师

岗位职责：
（以下一项或多项）：
1. 研究业务算法和AI算法（如 Transformer 、多模态模型），提取底层计算特征，评估对硬件的影响和需求；进行前沿算法的可行性研究和性能预测；
2. 负责NPU高性能算子库的设计与实现，确保关键算子（如 Attention）在硬件上的 100% 算力释放；
3. 与编译器团队协作，设计高效的算子到硬件的映射策略，包括算子融合、数据分块、流水线调度等，确保编译生成的代码能充分发挥硬件性能；设计算法到硬件的映射策略，负责全链路性能预算（Performance Budget）的分配与优化；
4. 撰写算子优化指南和最佳实践文档，建立算子性能数据库，为应用开发和系统集成提供清晰的参考。

任职要求：

1. 硕士及以上学历，计算机、电子工程相关专业；具有5年及以上高性能计算优化的工作经验；
2. 深入理解 CNN、Transformer、LLM、Diffusion、强化学习等主流算法的数学原理和计算特性；有过深度学习模型设计、训练或推理优化的实际项目经验的优先；
3. 深入理解 CPU/GPU/NPU 等加速器的微架构，包括缓存层次、指令流水线、向量执行单元等；能够进行硬件-软件协同优化；
4. 熟悉编译器原理（如 LLVM、TVM、MLIR），有过算子编译或代码生成的经验；理解指令级并行、向量化、循环优化等编译优化技术；
5. 精通性能分析工具（如 Nsight），能够通过性能计数器识别瓶颈；掌握 Python/C++/CUDA 等编程语言，有过高性能代码优化的经验；
6. 具备强烈的性能优化驱动力和数据驱动的思维；良好的学习能力，能够快速掌握新算法和新技术；优秀的团队协作和沟通能力。

【职位类别】芯片序列
【学历要求】硕士研究生及以上
【经验要求】5年
【招聘项目】日常招聘
【所属公司】北京地平线信息技术有限公司