返回岗位列表

寒武纪

编译器优化工程师

地点:北京 薪资:薪资未公开 日期:2025-09-24

岗位摘要

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

岗位职责

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

任职要求

  • 优秀的学习能力,对AI编译器和计算优化有浓厚兴趣
  • 熟悉C++/Python,具备编译器或AI编译优化开发经验,了解Triton、MHLO流程
  • 熟悉MLIR、Linalg等基础设施

加分项

  • 参与过 Triton、MLIR、IREE、XLA、TVM、TorchInductor 等编译器的开发,或在 GitHub 提交过相关 PR
  • 具备MLU/NPU/GPU 相关的编译优化经验
  • 有深度加速器上(gpu/npu)相关算子开发经验
  • 在 NeurIPS、ICLR、PLDI、CGO、MICRO 等学术会议上发表过 AI 编译相关论文