返回岗位列表

寒武纪

大模型训练软件开发工程师

地点:北京 薪资:30-60K 日期:2026-05-08

岗位摘要

参与大模型训练框架的设计、开发和优化工作;负责大模型预训练、后训练、强化学习等训练流程的工程化实现;进行多模态模型的训练算法研发与性能调优;设计并实现分布式训练系统,解决大规模并行计算中的技术难题

岗位职责

  • 参与大模型训练框架的设计、开发和优化工作
  • 负责大模型预训练、后训练、强化学习等训练流程的工程化实现
  • 进行多模态模型的训练算法研发与性能调优
  • 设计并实现分布式训练系统,解决大规模并行计算中的技术难题
  • 优化模型训练效率,提升计算资源利用率

任职要求

  • 本科及以上学历,1-3年相关工作经验
  • 熟练掌握Python和C++编程语言,具备良好的编码风格和工作习惯
  • 深入理解Transformer架构、自注意力机制等大模型核心技术
  • 熟悉预训练、后训练、强化学习、多模态等训练方法,掌握对应性能优化方法
  • 熟悉至少一种大模型训练框架,如Megatron-LM、Deepspeed、Transformers等
  • 具备分布式训练、并行计算、模型加速与性能优化经验
  • 熟悉机器学习、自然语言处理算法及算法工程化落地

福利待遇

  • 薪资范围30-60K,待遇优厚
  • 参与前沿大模型技术研发,接触行业顶尖技术
  • 工作地点位于北京海淀区致真大厦,交通便利
  • 本科及以上学历,1-3年经验即可投递,成长空间大

工作地址

北京海淀区致真大厦5楼