返回岗位列表

深度学习框架开发工程师

查看原岗位

地点：深圳薪资：30-60K 日期：2026-03-03

岗位摘要

参与PyTorch、PaddlePaddle等框架与摩尔线程软硬件栈的适配、训练与推理部署；研究并实现单卡及分布式训练中的并行、异构计算与性能优化；针对自研GPU设计模型压缩、量化、剪枝等优化方案并落地

岗位职责

参与PyTorch、PaddlePaddle等框架与摩尔线程软硬件栈的适配、训练与推理部署
研究并实现单卡及分布式训练中的并行、异构计算与性能优化
针对自研GPU设计模型压缩、量化、剪枝等优化方案并落地

任职要求

精通Linux、C++、Python，算法扎实，编程规范，系统设计能力强
深入理解CV/NLP模型，有TensorFlow/PyTorch等框架开发或优化经验
熟悉XLA、TVM、MLIR等深度学习编译器栈
熟悉参数服务器、多维并行、显存交换等分布式训练技术
具备模型量化、剪枝、蒸馏、NAS等压缩调优经验
熟练使用MPI、CUDA，有算子开发及并行编程经验
责任心强，学习沟通自驱力好，校招亦可

福利待遇

年薪30-60K×16薪，极具竞争力
深圳/杭州/北京多地点可选
参与国产GPU核心生态建设，技术成长空间大

工作地址

深圳南山区科兴科学园南山区科苑路15号科兴科学园