[北京]摩尔线程智能科技（上海）有限责任公司

职位：深度学习框架开发工程师
发布时间：2025-02-25
工作地点：北京
信息来源：浙江大学
职位类型：全职
职位描述
摩尔线程智能科技（上海）有限责任公司

其他企业 科学研究和技术服务业 上海市浦东新区

招聘信息

深度学习框架开发工程师

面议 北京市朝阳区 全职 本科及以上 2人 2025-06-01

收藏

2025-02-25 09:12:26

职位描述

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

熟悉Linux开发环境，精通C++和Python；

深入理解典型的深度学习算法（如CV和NLP模型）、主流深度框架（如TensorFlow、PyTorch、MxNet、Caffe、PaddlePaddle等）开发或优化经验；

有深度学习编译器（如XLA、TVM、MLIR）等软件栈的开发经验；

精通分布式训练技术，有参数服务器、多维并行、显存交换等技术的深入研究与开发者可优先考虑；

有机器学习模型性能调优和模型压缩（量化、剪枝、蒸馏、NAS等）经验者可优先考虑；

精通并行计算通信库（如MPI），有丰富的CUDA并行编程和典型机器学习算子开发经验者可优先考虑；

强烈的工作责任心，较好的学习能力、沟通能力和自驱力。

职位类别:计算机软、硬件/互联网/IT

专业要求:理学,工学

单位简介

摩尔线程智能科技成立于2020年10月，致力于创新面向元计算应用的新一代GPU，构建融合视觉计算、3D图形计算、科学计算及人工智能计算的综合计算平台，建立基于云原生GPU计算的生态系统，助力驱动数字经济发展。

联系方式

联系人：乔宇杰

联系电话：

电子邮箱：yujie.qiao@

传真：

公司地址

地址：上海市浦东新区

邮编：