职位详情
AI编译器设计
5-10万·15薪
广州泽尔马管理咨询有限公司
北京
5-10年
本科
11-03
工作地址

丽金智地中心

职位描述
工作职责
1.设计和开发高性能的编译器框架,支持多种异构计算设备(如CPU、GPU、NPU等)
2.负责基于NPU的编译器前端、中端、后端的设计和优化
3.基于mlir的处理前端图优化,后端流水,tiing等
4.10融合优化,图融合优化,其他NPU性能优化方式
5.多Cluster多核心调度,shape可变处理
6.Transformer,Actention的针对优化(大模型经验)--新增高阶专家角色
任职要求
1.本科及以上学历,计算机、软件相关专业,深入理解编译器理论与开发,有丰富的编译器架构设计经验;
2.有高性能计算和并行计算优化经验5年以上,熟悉相关算法和优化技巧。
3.熟悉编译原理、AI计算框架以及相关编译优化技术、AI加速;
4.熟悉C/C++/Python,熟悉C++/Python底层实现原理,C++编程经验5年以上。

加分项
1.熟悉arm,riscv,x86,gpu等常见体系结构;了解这些关键特性在编译器后端的设计和实现,并参与相关社区者优先;
2.有深度学习框架开发经验,熟悉TVM或MLIR中的一种,有实际开发经验;
3.有传统编译器(gcc,lvm),以及高性能算子库库开发经验者优先

工作地点:北京海淀区

以担保或任何理由索取财物,扣押证照,均涉嫌违法,请提高警惕

立即申请