AI编译器开发工程师

BeijingShanghaiChengduHangzhouFull-timePosted Dec 15, 2025

Job description

  1. 负责Triton、TileLang 等 GPU 领域专用语言(DSL)的编译框架研发与优化;

  2. 负责Triton/TileLang DSL 语法扩展与解析,支持新增 DSL 语法特性;

  3. 基于 MLIR/TVM/LLVM 构建 Triton/TileLang 专用 Dialect,开发优化Pass;

  4. 针对自研GPU芯片进行Triton及TileLang的编译性能优化,目标性能与手写算子一致;

  5. 分析 Triton/TileLang 生成代码的性能瓶颈,通过编译层面的内存访问模式优化、指令流水线调度、自动向量化等提升代码性能;

  6. 跟踪 Triton/TileLang 开源社区前沿技术,探索 DSL 编译创新方向,参与开源社区贡献。

  7. 本科及以上学历,计算机、高性能计算、微电子等相关专业,5年及以上编译器开发经验;

  8. 参与过 Triton/TileLang/MLIR 开源项目开发;

  9. 熟悉GPGPU架构及异构编程模型,具备较强问题分析与定位能力;

  10. 精通C++,CUDA及Python,具备模块化代码设计及调优能力;

  11. 良好的跨团队协作能力:能与 GPU 硬件架构工程师、算法工程师及框架工程师高效沟通协作;

  12. 加分项:有自研 GPU 的 Triton/TileLang 编译器后端开发经验,实现过核心算子的编译适配。

Description copied from 曦望Sunrise's careers page. Read the full posting before you apply.

More jobs at 曦望Sunrise

See all openings at 曦望Sunrise

AI编译器开发工程师 jobs at other companies