岗位描述(节选)
职位描述
我们正在寻找一位 编译器优化工程师,专注于在寒武纪产品上优化 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、……
完整岗位描述与官方投递入口,请登录后在岗位详情页查看。