返回岗位列表

寒武纪

大模型优化工程师

地点:上海 薪资:30-50K 日期:2026-03-03

岗位摘要

深度优化分布式训练框架,在性能、内存、并行策略等方面进行研究与开发,确保大规模训练场景下的效率和稳定性;针对大语言模型及视觉生成等场景,构建支持完整训练流程的解决方案,从数据预处理到训练工具构建进行全方位优化

岗位职责

  • 深度优化分布式训练框架,在性能、内存、并行策略等方面进行研究与开发,确保大规模训练场景下的效率和稳定性
  • 针对大语言模型及视觉生成等场景,构建支持完整训练流程的解决方案,从数据预处理到训练工具构建进行全方位优化
  • 密切关注业界最新技术动态,对前沿技术进行深度分析与研究,将技术成果融入分布式训练框架,推动框架持续迭代升级

任职要求

  • 计算机相关专业本科及以上学历,熟悉大模型训练框架 Megatron-LM 或 Deepspeed 等
  • 具备良好的编程能力,熟练掌握性能分析工具和方法,熟悉 Torch 的性能分析工具等
  • 逻辑思维清晰,具有良好的沟通能力、抗压能力和责任心
  • 具有相关开发或优化经验者优先考虑

工作地址

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