AI编译器研发工程师-27届当面告知
中科寒武纪科技股份有限公司
2027-02-07 21:29:11
校招27届本科硕士
岗位详情
地区：
北京市
·
上海市
专业要求：
专业不限
岗位职责：
研究和优化 Triton 编译器和基于MHLO的图编译器，使其高效支持 寒武纪芯片 计算架构；
基于 MLIR + Linalg 框架，提升编译器代码执行效率；
参与 图优化、后端代码生成、调度策略、指令优化、流水并行化相关优化工作；
研究和优化寒武纪芯片 上的 Kernel 计算性能，降低计算开销，提高吞吐率；
研究算子融合（Op Fusion）、自动调优（Auto-Tuning）、代码生成（CodeGen） 相关优化策略；
与 硬件团队、AI框架团队（如 PyTorch/TensorFlow） 协作，提升 寒武纪AI编译器对寒武纪 NPU 的支持能力；
跟踪 LLVM、Triton、TVM、XLA、TorchInductor 等 AI 编译器前沿技术，优化编译性能。
招聘要求：
计算机、电子工程、数学或相关专业，本科及以上学历；
具备扎实的编程基础，熟练掌握C++和Python；
对编译器技术有浓厚兴趣，熟悉MLIR、Linalg或类似编译框架者优先；
有Triton开发经验或参与过相关开源项目者优先；
有GPU/TPU/DSP等芯片的Kernel优化或性能调优经验者加分；
学习能力强，具备良好的问题解决能力和团队协作意识。
加分项：
参与过 Triton、IREE、XLA、TVM、TorchInductor 等编译器的开发，或在 GitHub 提交过相关 PR；
在校期间参与过编译器、AI框架或高性能计算相关项目；
有ACM、OI等竞赛背景或开源社区贡献；
在 NeurIPS、ICLR、PLDI、CGO、MICRO 等学术会议上发表过 AI 编译相关论文。
注：本岗位设置笔试环节，请留意邮件及短信通知。
公司介绍
中科寒武纪科技股份有限公司
中科寒武纪科技股份有限公司，成立于2016-03-15，位于北京，法定代表人：陈天石，登记状态：存续，注册资本41835.0224万人民币，公司类型：民营/私企
公司信息:
公司名称:中科寒武纪科技股份有限公司
公司类型:
公司性质:--
公司地点:
公司规模:
涉及领域:--
公司网址: