DSL编译器开发工程师(AI芯片方向)
Technology
成都市, 中国¥35,000 - ¥65,000 /月4周前截至 2026/12/7
全职
职位描述
该职位来源于猎聘 工作职责: 1.负责自研AI芯片Tile-based DSL编译器的后端设计与开发,构建高性能代码生成链路; 2.主导编译器中端优化管道建设,基于AST重写与动态追踪混合编译策略,提升IR生成效率及目标代码质量; 3.负责编译器IR在自研芯片上的功能验证与性能调优,确保编译产出与硬件微架构的高效匹配; 4.深度参与并优化基于该DSL的高性能算子库,重点覆盖通用矩阵乘(GEMM)、分组GEMM、融合多头注意力(FMHA)等核心算子; 5.与硬件架构团队紧密协同,参与芯片微架构设计评审,从编译视角驱动硬件特性优化。 任职要求: 1.计算机相关专业本科及以上学历,精通C++与Python,具备扎实的系统软件工程能力; 2.深入掌握至少一种主流DSL编程语言(如Triton、TileLang),具备编译器层面的开发或定制经验; 3.熟悉MLIR或TVM编译框架,了解LLVM后端基础设施,具备中间表示(IR)设计与降级实战经验; 4.熟悉GPGPU体系结构及CUDA编程模型,对计算机体系结构与编译原理有深刻理解; 5.了解Agent Kernel编程范式,具备异构计算场景下的调度优化意识。 加分项: 1.有计算机领域高水平编程竞赛(如ACM-ICPC、超算竞赛等)获奖经历; 2.具备CUDA/OpenCL/Triton等算子级手工调优经验,熟悉性能分析工具(如Nsight、VTune); 3.熟悉MLIR, TVM, LLVM优先; 4.有国产AI芯片编译器或算子库实际开发经验。
Keywords
BotanOpenCLVTune
对这个职位感兴趣吗?