AI编译器研发工程师当面告知
中科寒武纪科技股份有限公司
2027-01-03 22:16:49
其他社招
岗位详情
地区：
北京市
专业要求：
电子信息工程自动化
岗位职责：
我们正在寻找一位 编译器优化工程师，专注于在寒武纪产品上优化 Triton 编译器和基于MHLO的图编译器，以提升 AI 计算的性能和效率。你将负责 AI编译器在寒武纪架构上的适配和优化。
岗位职责
1.研究和优化 Triton 编译器和基于MHLO的图编译器，使其高效支持 寒武纪芯片 计算架构；
2.基于 MLIR + Linalg 框架，提升编译器代码执行效率；
3.参与 图优化、后端代码生成、调度策略、指令优化、流水并行化 相关优化工作；
4.研究和优化 Kernel 计算性能，降低计算开销，提高吞吐率；
5.研究算子融合（Op Fusion）、自动调优（Auto-Tuning）、代码生成（CodeGen） 相关优化策略；
6.与 硬件团队、AI框架团队（如 PyTorch/TensorFlow） 协作，提升AI编译器的支持能力；
7.跟踪 LLVM、Triton、TVM、XLA、TorchInductor 等 AI 编译器前沿技术，优化编译性能。
招聘要求：
1.优秀的学习能力，对 AI 编译器、AI计算优化有浓厚兴趣；
2.熟悉 C++/Python，具备编译器或 AI 编译优化开发经验，熟悉 Triton、MHLO 编译流程；
3.熟悉 MLIR、Linalg等基础设施；
4.具备ai编译优化（如 TVM、XLA、TorchInductor） 经验者优先；
5.熟悉 Kernel 计算优化、算子融合、自动并行化等高性能计算优化策略者加分。
加分项（优先考虑）
1.参与过 Triton、MLIR、IREE、XLA、TVM、TorchInductor 等编译器的开发，或在 GitHub 提交过相关 PR。
2.具备MLU/NPU/GPU 相关的编译优化经验。
3.有深度加速器上（gpu/npu）相关算子开发经验。
4.在 NeurIPS、ICLR、PLDI、CGO、MICRO 等学术会议上发表过 AI 编译相关论文。
公司介绍
中科寒武纪科技股份有限公司
中科寒武纪科技股份有限公司，成立于2016-03-15，位于北京，法定代表人：陈天石，登记状态：存续，注册资本41835.0224万人民币，公司类型：民营/私企
公司信息:
公司名称:中科寒武纪科技股份有限公司
公司类型:
公司性质:--
公司地点:
公司规模:
涉及领域:--
公司网址: