返回岗位列表

大模型训练优化工程师

查看原岗位

地点：上海薪资：28-55K 日期：2026-05-08

岗位摘要

负责客户AI模型在寒武纪MLU芯片上的端到端性能优化与精度调优，开发性能与精度分析工具；负责寒武纪Model Zoo的开发和维护，进行SOTA算法复现与性能优化，形成领域参考解决方案

岗位职责

负责客户AI模型在寒武纪MLU芯片上的端到端性能优化与精度调优，开发性能与精度分析工具
负责寒武纪Model Zoo的开发和维护，进行SOTA算法复现与性能优化，形成领域参考解决方案
针对公开或私有AI Benchmark进行端到端优化，跟踪AI平台厂商优化技术实践，最大化MLU在Benchmark上的表现

任职要求

EECS或相关专业本科或硕士以上学历
熟悉Megatron、DeepSpeed等分布式框架，有大规模分布式训练实际经验
对大模型、多模态、强化学习中至少一个场景有积累和见解
有GPU性能优化和精度调优经验，熟练掌握nvprof、nvvp、nsight等性能分析工具
熟练掌握TensorFlow/PyTorch等主流框架中的一种或以上
熟练操作Linux系统，精通C/C++/Python/Shell，具有扎实编程和调试基础
1年以上大模型性能优化和精度调试经验

加分项

（满足一项即可）
对数据/模型并行等分布式方案有一定的见解和经验
对一种或以上主流框架(TensorFlow/PyTorch等)有开发者视角的理解，对框架设计或者调优有一定的见解和经验
熟悉并喜欢高性能优化，对CUDA/OpenCL/BANG/C/OpenMP等并行计算模型有一定的开发经验
熟悉GPGPU/MLU/CPU微架构，对软硬件联合优化有一定的见解和经验
陶先生刚刚活跃

福利待遇

有竞争力的薪酬体系，28-55K·15薪
参与寒武纪国产AI芯片前沿技术研发
大模型与高性能计算双重领域深耕机会
技术氛围浓厚，职业发展通道畅通

工作地址

上海浦东新区展想广场1号楼1805