返回岗位列表

AI芯片高性能计算库开发工程师

查看原岗位

地点：上海薪资：薪资未公开日期：2026-01-29

岗位摘要

基于寒武纪AI芯片以及硬件平台，开发高性能计算库；在寒武纪高性能异构平台上，分析算子中可能存在的性能瓶颈，提出性能优化策略并达成优化目标；负责算子的需求分析、设计、开发、单元测试、优化、集成和版本维护

岗位职责

基于寒武纪AI芯片以及硬件平台，开发高性能计算库
在寒武纪高性能异构平台上，分析算子中可能存在的性能瓶颈，提出性能优化策略并达成优化目标
负责算子的需求分析、设计、开发、单元测试、优化、集成和版本维护
参与硬件验证、性能评估等相关工作
参与前沿技术的跟踪与分析

任职要求

计算机、电子工程、数学、通信、自动化等相关专业，硕士及以上学历
熟悉并行/异构计算体系结构，在TPU/NPU/GPU/ARM/X86/DSP等任意平台上拥有2年以上的开发和优化经验
熟悉Pytorch、Tensorflow框架及自定义算子开发，熟悉LLM常见模型结构
熟悉C/C++、Python、Shell，具备良好编程习惯，熟悉软件开发流程
学习能力强，主动思考，逻辑思维强，做事踏实、认真，具备一定抗压能力
具有良好的团队协作精神和沟通能力，能够独立解决技术难题