返回岗位列表

寒武纪

C++开发实习生(AI编译器)

地点:北京 薪资:250-300元/天 日期:2026-05-08

岗位摘要

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

岗位职责

  • 研究和优化 Triton 编译器及基于 MHLO 的图编译器,使其高效支持寒武纪芯片计算架构
  • 基于 MLIR + Linalg 框架,提升编译器代码执行效率
  • 参与图优化、后端代码生成、调度策略、指令优化、流水并行化等相关优化工作
  • 研究和优化 Kernel 计算性能,降低计算开销,提高吞吐率
  • 研究和优化算子融合(Op Fusion)、自动调优(Auto-Tuning)、代码生成(Code Gen)等相关策略

任职要求

  • 硕士及以上学历,数学、物理、计算机、微电子、通信、自动化等相关专业,2027届毕业生
  • 具备优秀的学习能力,对 AI 编译器、AI 计算优化有浓厚兴趣
  • 熟悉 C++/Python,具备编译器或 AI 编译优化开发经验,熟悉 Triton、MHLO 编译流程
  • 熟悉 MLIR、Linalg 等基础设施,具备 AI 编译优化(如 TVM、XLA、TorchInductor)经验者优先
  • 熟悉 Kernel 计算优化、算子融合、自动并行化等高性能计算优化策略者加分

福利待遇

  • 日薪 250-300 元,薪酬具有竞争力
  • 硕士及以上学历要求,岗位专业性强
  • 可深入参与 AI 编译器核心研发,积累前沿技术经验

工作地址

北京海淀区致真大厦D1601