NPU算子开发工程师/专家

岗位职责
负责公司自研NPU算子加速库的研发与维护。
深入分析AI模型在NPU上的性能瓶颈，针对自研芯片架构进行高性能计算加速与优化。
与编程模型工具链团队深度协作，相互迭代演进以提高算子的开发调优效率，提高算子实现的跨产品可迁移性。
协同芯片设计团队，基于硬件架构特点进行软硬协同调优，深度参与后续芯片产品的软硬件协同设计。

任职要求
必备条件：
熟练掌握C++/Python，编程基础扎实。
熟悉计算机体系结构及操作系统原理，理解指令集、流水线、存储层次等核心概念。
熟悉深度学习算法，对主流深度学习框架（PyTorch/vLLM等）有应用或开发经验。
熟悉大算力GPU或NPU的硬件架构，编程模型，有丰富的AI算子开发经验。
具备优秀的分析问题和解决问题的能力。
技术栈要求（满足其一即可）：
有CUDA或主流NPU厂商对应软件栈的开发或深度使用经验。
有Triton/TileLang/Cutile/Cutlass等开发或深度使用经验。
优先条件：
有AI算子开发的项目管理或团队管理经验。
有大模型典型融合算子从性能分析，方案设计，工程实施到性能验证的端到端经验。
有NPU/Tensor Core等AI加速单元底层开发经验。
熟悉TVM、MLIR等深度学习编译器，有算子自动生成经验。
有AI芯片Post-silicon Bring-up或性能调优经验。
计算机领域编程大赛获奖者或有相关顶会论文发表。