独角兽

AI编译器工程师

此芯科技

开发社区人工智能社会招聘

岗位描述

岗位描述:使用 AI 编译器技术,在 RISCV-AI 架构及未来计算架构上,以自动化、可交互的方式生成高效的计算密集、访存密集型、分布式算子;与硬件、算法、方案团队紧密合作,共同解决技术难题,持续推动 RISCV-AI 计算架构及编程范式的演进与创新。负责 AI 编程语言设计与开发,包括但不限于 Triton、TileLang、CuTile 等 Python 高性能编程语言的设计与优化开发,探索 RISCV-AI 架构的编程范式;负责 AI 编译器中后端开发,基于对算子 Tiling、访存分析、指令调度、硬件微架构等认知,以 IR 编译方式优化算子和计算图,开发支撑 AI 编译算子、计算图运行的 Runtime;基于对并行计算原理的深刻理解参与 RISCV-AI 架构升级;与算法和应用团队密切配合,了解业务需求,为 AI 框架的功能和性能提供支持;持续关注业界 AI 框架和编译技术的发展动态,参与公司技术储备和前沿探索。 岗位要求:计算机、电子工程及相关专业本科及以上学历,熟悉 C/C++;熟悉 IREE、Triton、OpenXLA 等 MLIR 编译生态者优先;对计算机体系结构、深度学习加速器等硬件有较深入了解和工程经验;具备良好的问题分析和解决能力,能够独立承担复杂任务并交付高质量结果;具有出色的沟通和团队协作能力,能与跨职能团队高效配合;对前沿技术保持持续关注和学习热情,具有创新精神和开放心态。