高性能算法库工程师当面告知
中科寒武纪科技股份有限公司
2027-01-03 22:16:56
社招博士硕士
岗位详情
地区:
上海市
专业要求:
数学与应用数学电子信息工程自动化计算机科学与技术软件工程
岗位职责:
1. 基于寒武纪AI芯片以及硬件平台,开发高性能计算库;
2. 在寒武纪高性能异构平台上,分析算子中可能存在的性能瓶颈,提出性能优化策略并达成优化目标;
工作包括但不限于针对算子的需求分析、设计、开发及单元测试、优化、集成和版本维护;
3. 参与硬件验证、性能评估等相关工作;
4. 参与前沿技术的跟踪与分析。
招聘要求:
1. 计算机、电子工程、数学、通信、自动化等相关专业,硕士及以上学历;
2. 熟悉并行/异构计算体系结构,有在TPU/NPU/GPU/ARM/X86/DSP等任意平台上2年以上的开发和优化经验;
3. 熟悉 Pytorch、Tensorflow 框架及自定义算子开发,熟悉LLM常见模型结构;
4. 熟悉 C/C++、python、shell,具备良好编程习惯,熟悉软件开发流程;
5. 学习能力强,主动思考,逻辑思维强,做事踏实、认真,具备一定抗压能力;
6. 具有良好的团队协作精神和沟通能力,能够独立解决技术难题。
公司介绍

中科寒武纪科技股份有限公司
中科寒武纪科技股份有限公司,成立于2016-03-15,位于北京,法定代表人:陈天石,登记状态:存续,注册资本41835.0224万人民币,公司类型:民营/私企
公司信息:
公司名称:中科寒武纪科技股份有限公司
公司类型:
公司性质:--
公司地点:
公司规模:
涉及领域:--
公司网址: