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