Ivyea Jobs 9543 roles · 169 companies · 刚刚
曦望 · AI 芯片

AI编译器开发工程师

北京 / 上海 / 成都 / 杭州 社招 · 全职 推理 / 部署 工程开发

职位描述

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 编译创新方向,参与开源社区贡献。

任职要求

1. 本科及以上学历,计算机、高性能计算、微电子等相关专业,5年及以上编译器开发经验; 2. 参与过 Triton/TileLang/MLIR 开源项目开发; 3. 熟悉GPGPU架构及异构编程模型,具备较强问题分析与定位能力; 4. 精通C++,CUDA及Python,具备模块化代码设计及调优能力; 5. 良好的跨团队协作能力:能与 GPU 硬件架构工程师、算法工程师及框架工程师高效沟通协作; 6. 加分项:有自研 GPU 的 Triton/TileLang 编译器后端开发经验,实现过核心算子的编译适配。
曦望去官方页面投递 →