NPU 算子库开发工程师

该职位来源于猎聘 职位描述 算子开发与维护：在导师指导下参与 NPU 算子库的开发与维护，学习算子库整体架构、接口标准与开发流程，实现基础 AI 算子的功能 精度与性能验证：编写算子的单元测试、精度测试与性能测试用例，保障算子的计算精度与基础性能达标 性能优化协助：协助资深工程师完成核心算子的性能调优，学习 pingpang buffer、算子融合、矢量化加速、多核调度等优化手段 前沿技术预研：学习大模型核心算子（FlashAttention、Matmul、LayerNorm 等）的优化技术，参与新技术的预研与验证工作 任职要求 本科及以上学历，计算机、电子工程、自动化、数学、软件工程等相关专业 掌握 C/C++ 编程语言，了解 Python，有良好的代码规范与问题排查能力 对计算机体系结构、高性能计算、AI 算子的计算逻辑有基础认知 具备极强的动手能力与学习能力，对底层硬件与性能优化有浓厚兴趣 有良好的团队协作意识，能配合团队完成开发与测试任务 加分项 有 CUDA/OpenCL 编程经验，或写过简单的 GPU/NPU 算子 有 PyTorch/TensorFlow 等深度学习框架的使用经验，了解 AI 模型的计算流程 有高性能计算、算法竞赛（ACM等）相关经历 了解量化、剪枝等模型压缩技术，或有 NPU/GPU 硬件相关的学习经历