返回岗位列表

深度学习框架工程师

查看原岗位

地点：北京薪资：25-50K 日期：2026-03-03

岗位摘要

参与PyTorch、PaddlePaddle等主流框架与摩尔线程软硬件栈对接，完成模型训练、推理与部署；研究并实现单卡及分布式训练技术，包括并行策略、异构计算及性能优化；针对自研GPU设计模型压缩与加速方案，持续降低训练与推理成本

岗位职责

参与PyTorch、PaddlePaddle等主流框架与摩尔线程软硬件栈对接，完成模型训练、推理与部署
研究并实现单卡及分布式训练技术，包括并行策略、异构计算及性能优化
针对自研GPU设计模型压缩与加速方案，持续降低训练与推理成本

任职要求

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

福利待遇

25-50K×16薪，极具竞争力的薪酬包
零距离参与万卡集群与基座大模型核心技术攻关
弹性工作、技术分享、顶级GPU算力资源
望京国际研发园现代化办公环境，团队氛围开放包容

工作地址

北京朝阳区望京国际研发园l座3层