返回岗位列表

AI深度学习框架技术总监

查看原岗位

地点：北京薪资：60-90K 日期：2026-05-08

岗位摘要

作为技术Leader参与PyTorch、PaddlePaddle等主流机器学习框架与摩尔线程软硬件栈的对接，完成模型训练、推理及部署工作；负责单GPU及分布式训练技术的研究与实现，设计开发各类并行与异构计算方案并进行性能优化

岗位职责

作为技术Leader参与PyTorch、PaddlePaddle等主流机器学习框架与摩尔线程软硬件栈的对接，完成模型训练、推理及部署工作
负责单GPU及分布式训练技术的研究与实现，设计开发各类并行与异构计算方案并进行性能优化
研究并实现适用于摩尔线程自研GPU的模型优化与压缩方法，提升模型运行效率
参与深度学习编译器相关技术栈的研发与优化工作

任职要求

熟悉Linux开发环境，精通C++和Python，具备扎实的算法基础、良好的编程风格和系统设计能力
深入理解CV和NLP等典型深度学习算法，有TensorFlow、PyTorch、MxNet、Caffe、PaddlePaddle等主流框架的开发或优化经验
有深度学习编译器（如XLA、TVM、MLIR等）软件栈的开发经验
精通分布式训练技术，对参数服务器、多维并行、显存交换等有深入研究
有机器学习模型性能调优或模型压缩（量化、剪枝、蒸馏、NAS等）经验者优先
精通MPI等并行计算通信库，有CUDA并行编程及典型机器学习算子开发经验者优先
有优秀论文发表或算法竞赛获奖经历者优先
有参与开源项目经历者优先
有强烈的工作责任心，良好的学习能力、沟通能力和自驱力

福利待遇

提供具有竞争力的薪酬体系，薪资范围60-90K，16薪
硕士及以上学历，5-10年相关工作经验
参与前沿AI硬件与软件协同优化项目
工作地点位于北京朝阳区望京国际研发园

工作地址

北京朝阳区望京国际研发园I