深圳-AI异构计算工程师(J86070)
  • 招聘类别:
  • 校园招聘
  • 工作性质:
  • 全职
  • 薪资范围:
  • 面议
  • 招聘人数:
  • 6
  • 发布时间:
  • 2026-05-01
  • 截止时间:
  •  
  • 工作地点:
  • 广东省-深圳市

工作职责:

-负责飞桨深度学习平台( PaddlePaddle)核心框架研发,支撑文心大模型训推技术迭代;
-构建全球领先的AI异构计算加速引擎和加速平台,建立软硬件一体全国产化解决方案;
-负责飞桨深度学习框架底层调度优化、高性能算子开发和软硬件适配前沿技术探索;
-负责面向CPU/GPU/ASIC等多元化计算架构的通用编译系统开发、编译优化等前沿技术突破;
-负责异构高性能计算平台的设计、研发,高性能计算库、通信库开发与优化;
-负责飞桨深度学习框架核心架构升级和框架体验优化,提升框架生态兼容能力。


任职资格:

-热爱编程和深度学习框架架构技术,精通C++/Python,具备快速学习能力;
-具有独立开发能力,对AI算法和主流框架有丰富的应用和二次开发经验;
-精通GPU/ARM/MIPS/DSP等任意异构计算平台;
-精通计算机体系结构,有汇编级别开发经验;
-有探索精神和出色的团队协作及沟通能力;
具有以下条件者优先:
-精通PaddlePaddle、Pytorch、TensorFlow等开源框架,做过源码级优化移植等工作;
-对分布式计算有深入理解,特别是通信策略在AI计算中的应用;
-熟练使用Cublas、Cudnn、MIopen、OpenBlas、MKL、Eigen等主流计算库;
-精通CUDA、OpenCL和ARM-GPU等开发,有过并行计算优化协同开发经验;
-熟悉AI training通信过程,熟悉MPI、NCCL、RDMA、GPU Direct等通信技术;
-精通常用硬件平台性能分析工具链,如CodeXL\NVVP\GPA等;
-精通LLVM、SVM等编译优化原理和具备相关实践经验;
-有Linux内核相关开发和优化经验。

©2026  百度昆仑芯定制网申   京ICP备05051632号-16 京公网安备 11010802032024号隐私政策 Powered by