Ivyea Jobs 8332 roles · 120 companies · 刚刚
算苗科技 · AI 芯片

AI芯片算子开发工程师

北京 社招 · 全职 推理 / 部署优化

职位描述

1.基于自研芯片架构开发大模型推断、训练、微调算子; 2.与编译器团队紧密配合,针对硬件体系结构对算法进行分析,设计,和优化; 3.将算子实现映射到实际硬件,进行性能分析,识别性能瓶颈; 4.算子库的软件设计、实现及优化,包括但不限于算子库实现流程本身的优化,指令级并行等,通过优化最大程度发挥硬件性能。

任职要求

1.精通C/C++语言编程; 2.熟悉Pytorch、Tensorflow框架及自定义算子开发,熟悉LLM常见模型结构; 3.熟悉计算机体系结构、众核处理器原理,具有算子性能优化的实际工作经验; 4.有良好的沟通能力和团队协作能力,优秀的学习能力和技术热情。 加分项: 1.具备SIMD,CUDA,ROCm,Ascend算子开发经验; 2.具备汇编指令级并行开发经验。
算苗科技去官方页面投递 →