返回岗位列表

AI编译器研发工程师

查看原岗位

地点：北京薪资：30-60K 日期：2026-03-03

岗位摘要

研究和优化Triton编译器及基于MHLO的图编译器，高效支持寒武纪芯片计算架构；基于MLIR + Linalg框架，提升编译器代码执行效率；参与图优化、后端代码生成、调度策略、指令优化、流水并行化相关优化工作

岗位职责

研究和优化Triton编译器及基于MHLO的图编译器，高效支持寒武纪芯片计算架构
基于MLIR + Linalg框架，提升编译器代码执行效率
参与图优化、后端代码生成、调度策略、指令优化、流水并行化相关优化工作
研究和优化寒武纪芯片上的Kernel计算性能，降低计算开销，提高吞吐率
研究深度学习算子融合、自动调优、代码生成相关优化策略
与硬件团队、深度学习框架团队协作，提升寒武纪AI编译器对NPU的支持能力
跟踪LLVM、Triton、TVM、XLA、TorchInductor等AI编译器前沿技术，优化编译性能

任职要求

计算机、电子工程、数学或相关专业硕士学历
具备扎实的编程基础，熟练掌握C++和Python
对编译器技术有浓厚兴趣，熟悉MLIR、Linalg或类似编译框架者优先
有Triton开发经验或参与过相关开源项目者优先
有GPU/TPU/DSP等芯片的Kernel优化或性能调优经验者加分
学习能力强，具备良好的问题解决能力和团队协作意识
参与过Triton、IREE、XLA、TVM、TorchInductor等编译器开发或在GitHub提交过相关PR者优先
有ACM、OI等竞赛背景或开源社区贡献者优先
在NeurIPS、ICLR、PLDI、CGO、MICRO等学术会议上发表过AI编译相关论文者优先

加分项

参与过 Triton、IREE、XLA、TVM、TorchInductor 等编译器的开发，或在 GitHub 提交过相关 PR
参与过编译器、AI框架或高性能计算相关项目
有ACM、OI等竞赛背景或开源社区贡献
在 NeurIPS、ICLR、PLDI、CGO、MICRO 等学术会议上发表过 AI 编译相关论文
王天一半年前活跃
寒武纪 · 高性能计算研发经理

福利待遇

具有竞争力的薪酬待遇：30-60K·15薪

工作地址

北京海淀区致真大厦D座11层