返回岗位列表

高性能算法库工程师

查看原岗位

地点：西安薪资：25-50K 日期：2026-03-03

岗位摘要

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

岗位职责

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

任职要求

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

工作地址

西安雁塔区新长安广场C座沣惠南路34号新长安广场二期C座2401