岗位职责
1、负责为新型AI芯片定制或优化编译器(如LLVM/GCC后端),支持该新型芯片的专用指令集(如AI矩阵运算指令),实现计算图、AI框架(PyTorch/TensorFlow)/模型到硬件指令的映射;
2、开展算子融合、图、自动向量化优化等工作
3、开展编译器的测试及BUG修复工作;
4、撰写相关方案、报告及各类文档;
5、领导交办的其他工作任务。
能力要求
1.工作经验:具备独立完成编译器研发及调优的开发工作经验;具有AI编译器实际开发经验者优先。
2.能力要求:
(1)熟悉编译器架构(IR优化、代码生成)、汇编语言或常见芯片指令集;
(2)精通LLVM/MLIR框架,有TVM/XLA等AI编译器实战经验;
(3)熟悉AI模型量化(INT8/FP16)和剪枝技术;
(4)工作积极主动,具备较强的责任心和原则性,具有良好的学习能力和抗压能力。