工作职责:
1、负责深度学习训练端编译器构造及编译优化相关技术研发工作
2、负责实现算子融合、循环优化、自动调度、自动调优等等后端优化技术
3、负责实现编译器与训练框架(如Pytorch,Tensorflow,Jax等)的集成
4、负责实现编译器与多款异构AI芯片的适配与性能优化
任职要求:
1、计算机或相关专业硕士及以上学历,机器学习框架或编译器相关的研究或工作背景优先
2、具备扎实的C++编程基本功,有 C++ 编译器(如LLVM,MLIR)和深度学习框架开发经验者优先
3、了解计算机体系结构及量化性能分析方法,了解GPU和加速器架构特点
4、有高性能AI算子开发经验者优先
欢迎咨询!
电话:18519274080
微信号:Brylin1991
邮箱: herocanjob@163.com
--
FROM 122.190.149.*