岗位描述(节选)
岗位职责:
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. ……
完整岗位描述与官方投递入口,请登录后在岗位详情页查看。