岗位描述
使用 AI 编译器技术在 RISCV-AI 架构及未来计算架构上,以自动化、可交互的方式生成高效的计算密集型、访存密集型和分布式算子;与硬件、算法、方案团队紧密合作,共同解决技术难题,持续推动 RISCV-AI 计算架构及编程范式的演进与创新;负责 AI 编程语言设计与开发,包括 Triton、TileLang、CuTile 等 Python 高性能编程语言的设计与优化,探索 RISCV-AI 架构的编程范式;负责 AI 编译器中后端开发,基于对算子 Tiling、访存分析、指令调度、硬件微架构等认知,以 IR 编译方式优化算子和计算图,并开发支撑 AI 编译算子、计算图运行的 Runtime;基于对并行计算原理的理解参与 RISCV-AI 架构升级;与算法和应用团队密切配合,了解业务需求,为 AI 框架的功能和性能提供支持;持续关注业界 AI 框架和编译技术的发展动态,参与公司技术储备和前沿探索。