返回岗位列表

深度学习框架开发工程师

查看原岗位

地点：杭州薪资：30-60K 日期：2026-03-03

岗位摘要

参与主流机器学习框架与摩尔线程软硬件栈的对接，完成模型训练、推理与部署；研究并实现单GPU及分布式训练技术，包括并行与异构计算设计、开发与性能优化；研究并实现面向摩尔线程自研GPU的模型优化与压缩方法

岗位职责

参与主流机器学习框架与摩尔线程软硬件栈的对接，完成模型训练、推理与部署
研究并实现单GPU及分布式训练技术，包括并行与异构计算设计、开发与性能优化
研究并实现面向摩尔线程自研GPU的模型优化与压缩方法

任职要求

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

福利待遇

年薪16薪，30-60K，极具竞争力
杭州滨江核心地段办公环境
参与国产GPU大模型核心生态建设
技术成长快，团队大牛多，校招友好

工作地址

杭州滨江区医惠中心A座9楼