首页 >> 招聘信息 >>招聘 >> 算子开发工程师
详细内容

算子开发工程师

岗位职责:
1. 利用深度学习编译技术,结合专家级经验(算法/系统/硬件),解决搜广推、CV、NLP,以及多模态、大模型的训练和推理场景挑战。
2. 系统化分析模型,提炼/抽象优化方法,寻求解空间(包括算子调度,融合,Schedule,异构调度等)。
3. multi-framework(例如Tensorflow、Pytorch等) 计算图优化Pass开发,手段包括不限于编译IR优化,专家经验模板化等等。
4. multi-device(例如 GPU,CPU,NPU)专项 SIMT、SIMD 计算加速,包括核心 Gemm 算子,Reduce算子,以及复杂融合算子等等。
岗位基本要求:
有扎实的计算机理论基础,熟练掌握C++和Python语言
具备以下优先:
1. 熟悉PyTorch 或 TensorFlow等引擎的架构和运行原理,有深度定制经验优先
2. 熟悉TVM/MLIR/XLA等DL编译器,有Cutlass技术经验,具备计算图翻译至IR并优化的经验优先
3. 在深度学习领域有丰富的模型优化经验,熟悉GPU或NPU架构,有CUDA或NPU编程经验优先
4. 学习能力和好奇心强,具备良好的问题分析和解决能力,有团队合作精神和沟通能力
工作地址:
苏州:苏州新区塔园路101号悦峰大厦
杭州:萧山区利一路天人大厦
上海:浦东新区盛夏路399弄亚芯科技园
成都:武侯区天府三街新希望国际

邮箱:recruiter@yizhu-tech.com
媒体垂询:pr@yizhu-tech.com
亿铸小助手微信:Yizhu_assistant

地址:苏州高新区塔园路101号佳兆业悦峰大厦19楼

网站导航