DSL编译器开发工程师
Technology
¥35,000 - ¥65,000 /月1个月前
全职
职位描述
该职位来源于猎聘 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
对这个职位感兴趣吗?