深度学习框架/分布式训练工程师

查看原岗位

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

岗位摘要

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

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

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