返回岗位列表

深度学习框架开发架构师

查看原岗位

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

岗位摘要

参与PyTorch、PaddlePaddle等主流框架与摩尔线程软硬件栈的适配，完成模型训练、推理与部署；负责单卡及分布式训练技术研究与实现，设计并行、异构计算方案并持续优化性能；研究并实现面向自研GPU的模型优化与压缩算法，提升计算效率与部署效果

岗位职责

参与PyTorch、PaddlePaddle等主流框架与摩尔线程软硬件栈的适配，完成模型训练、推理与部署
负责单卡及分布式训练技术研究与实现，设计并行、异构计算方案并持续优化性能
研究并实现面向自研GPU的模型优化与压缩算法，提升计算效率与部署效果

任职要求

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

福利待遇

年薪60-90K·16薪，极具竞争力
可灵活选择杭州/北京/深圳/上海办公
参与国产GPU核心生态建设，技术前沿成长快
望京国际研发园优质办公环境

工作地址

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