深度学习框架研发工程师
岗位摘要
参与主流机器学习框架（如PyTorch、PaddlePaddle）与摩尔线程自研GPU软硬件栈的对接工作，涵盖模型训练、推理与部署全流程；参与单GPU及分布式训练技术的研究与实现，包括各类并行与异构计算技术的设计、开发及性能优化
岗位职责
- 参与主流机器学习框架（如PyTorch、PaddlePaddle）与摩尔线程自研GPU软硬件栈的对接工作，涵盖模型训练、推理与部署全流程
- 参与单GPU及分布式训练技术的研究与实现，包括各类并行与异构计算技术的设计、开发及性能优化
- 研究并实现适合摩尔线程自研GPU的模型优化与压缩方法
任职要求
- 熟悉Linux开发环境，精通C++和Python，具备扎实的算法基础、良好的编程风格和系统设计能力
- 深入理解典型深度学习算法（如CV和NLP模型），具备主流深度学习框架（如TensorFlow、PyTorch、MxNet、Caffe、PaddlePaddle等）的开发或优化经验
- 具有深度学习编译器（如XLA、TVM、MLIR）等软件栈的开发经验
- 精通分布式训练技术，有参数服务器、多维并行、显存交换等技术深入研究与开发经验者优先
- 具备机器学习模型性能调优及模型压缩（量化、剪枝、蒸馏、NAS等）经验者优先
- 精通并行计算通信库（如MPI），具有丰富CUDA并行编程及典型机器学习算子开发经验者优先
- 具有强烈的工作责任心，较好的学习能力、沟通能力和自驱力
- 硕士及以上学历，5至10年相关工作经验
工作地址
北京朝阳区望京国际研发园I座