编译器优化工程师
岗位摘要
研究和优化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 编译相关论文