AI编译器研发工程师

岗位职责： 研究和优化 Triton 编译器和基于MHLO的图编译器，使其高效支持 寒武纪芯片 计算架构。 基于 MLIR + Linalg 框架，提升编译器代码执行效率。 参与 图优化、后端代码生成、调度策略、指令优化、流水并行化 相关优化工作。 研究和优化 Kernel 计算性能，降低计算开销，提高吞吐率。 研究算子融合（Op Fusion）、自动调优（Auto-Tuning）、代码生成（CodeGen） 相关优化策略。 与 硬件团队、AI框架团队（如 PyTorch/TensorFlow） 协作，提升AI编译器的支持能力。 跟踪 LLVM、Triton、TVM、XLA、TorchInductor 等 AI 编译器前沿技术，优化编译性能。 岗位要求： 优秀的学习能力，对 AI 编译器、AI计算优化有浓厚兴趣。 熟悉 C++/Python，具备编译器或 AI 编译优化开发经验，熟悉 Tr…… 完整岗位描述与官方投递入口,请登录后在岗位详情页查看。