返回岗位列表

NPU平台高性能计算库开发工程师

查看原岗位

地点：北京薪资：薪资未公开日期：2026-01-29

岗位摘要

完成高性能算子的需求分析与分解，满足NPU的各种应用场景；与芯片/架构团队紧密配合，持续优化算子性能，逐步完善软硬件架构；完成各类算子的软件方案设计及开发；分析算子/网络的性能瓶颈，提出性能优化策略，达成性能目标

岗位职责

完成高性能算子的需求分析与分解，满足NPU的各种应用场景
与芯片/架构团队紧密配合，持续优化算子性能，逐步完善软硬件架构
完成各类算子的软件方案设计及开发
分析算子/网络的性能瓶颈，提出性能优化策略，达成性能目标

任职要求

图像处理、模式识别、机器学习等相关专业硕士及以上学历
精通C++/Python软件编程语言
熟悉Onnx/TensorFLow/Pytorch一种主流神经网络框架，熟悉各类算子
熟悉并行/异构计算体系结构，在TPU/NPU/GPU一种平台有2年以上的开发和优化经验
具备业界常用高性能库(TensorRT/cuDNN/CUDA/OPENCV)开发和性能优化经验者优先
熟悉指令流水，计算机体系结构
良好的团队沟通能力、技术攻关能力，责任心强