返回岗位列表

大模型算法应用工程师

查看原岗位

地点：上海薪资：25-50K 日期：2026-05-08

岗位摘要

负责客户AI模型在寒武纪人工智能芯片(MLU)上的端到端性能优化和精度调优，及时总结可泛化的实践，开发性能/精度分析工具；负责寒武纪Model Zoo的开发和维护，在寒武纪人工智能芯片上进行SOTA算法复现，并进行性能优化分析和精度调优，形成领域参考解决方案

岗位职责

负责客户AI模型在寒武纪人工智能芯片(MLU)上的端到端性能优化和精度调优，及时总结可泛化的实践，开发性能/精度分析工具
负责寒武纪Model Zoo的开发和维护，在寒武纪人工智能芯片上进行SOTA算法复现，并进行性能优化分析和精度调优，形成领域参考解决方案
针对公开的或私有的AI Benchmark（如MLPerf等）进行端到端优化，密切跟踪评估各AI平台厂商的优化技术实践，最大化MLU在这些benchmark上的表现

任职要求

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

加分项

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

福利待遇

有竞争力的薪酬（25-50K·15薪）
知名AI芯片公司寒武纪平台，接触前沿技术
上海浦东核心办公地段
接触大规模分布式训练与芯片级性能优化核心业务

工作地址

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