【社招】【FQ科技】资深AI 编译工程师, 50-80k, 北京/上海
工作职责:
1、面向自研的AI处理器架构,交付高性能的算子库和通信库,并配合框架团队完成算子库向PyTorch的集成;
2、构建python级的算子编程模型和编译器,支持用户开发高性能的自定义算子;
3、参与完整的产品开发周期,包括硅前验证、硅后验证以及软件SDK的版本发布和维护。
岗位条件:
1、计算机/电子工程专业,硕士及以上(博士优先);
2、5年及以上AI算子开发和优化经验,对GPGPU或者其它AI加速器的编程模型有较深的理解;
3、核心技术:
(1)掌握编译原理的基本知识,尤其是编译器后端架构相关的pass设计和实现原理。能够在典型的AI 编译器(如TVM, Triton)中实现特定功能的pass;
(2)熟练掌握MLIR,能够使用常见dialect 完成从计算图级别的抽象到ISA级别抽象的lowering过程;
(3)了解AI处理器的ISA级特性,具有汇编级别的编程能力。针对特定的架构以及汇编代码能够进行性能分析、评估和指定优化方案;
(4)精通C/C++, Python编程,精通数据结构和算法的基本知识,具备良好的软件架构意识和代码质量意识。
4、具有一定的项目管理经验和知识储备;
5、良好的工作习惯和沟通表达能力;
6、关键加分项:
(1)有Trtion/Triton-lang/Triton-distributed等AI编译项目的开发经验;
(2)分布式系统中的通信算子开发和优化经验;
(3)有创新思维意识和优秀的逻辑分析能力,能够结合既往项目经验和具体问题并给出合理的分析和解答。
公司介绍:**科技有限公司成立于 2022 年,主攻方向为面向人工智能的多核 NPU 计算芯片,目前第一代产品即将流片
联系邮箱:bjbwhr@126.com
--
FROM 118.247.80.*