返回岗位列表

AI基础设施研发工程师

查看原岗位

地点：上海薪资：27-50K 日期：2026-03-03

岗位摘要

参与PyTorch、PaddlePaddle等主流机器学习框架与摩尔线程软硬件栈的对接、训练、推理与部署；实现推理框架及深度学习算子的性能优化；研究并实现单GPU及分布式训练技术，包括并行与异构计算的设计、开发与性能调优

岗位职责

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

任职要求

熟悉Linux开发环境，精通C++与Python，算法基础扎实，编程规范，具备系统设计能力
深入理解CV、NLP等典型深度学习算法，有主流框架开发或优化经验
具备深度学习编译栈（XLA、TVM）开发经验者优先
精通分布式训练技术，深入研究与开发过参数服务器、多维并行、显存交换者优先

福利待遇

年薪27-50K×16薪，极具竞争力
上海核心科技园区办公，交通便利
参与前沿GPU与AI框架深度整合项目，技术成长空间大
优秀开源项目经历者将获得额外认可与资源支持

工作地址

上海浦东新区上海长泰广场C栋