返回岗位列表

寒武纪

分布式训练研发工程师

地点:北京 薪资:25-35K 日期:2026-03-03

岗位摘要

基于MLU构建异构并行通信库,实现高性能设备通信加速库;面向未来低时延、高带宽的计算通信需求进行技术攻关;负责大规模分布式训练中多机计算、通信和训练性能的优化;本科及以上学历,计算机、软件工程、通信等相关专业

岗位职责

  • 基于MLU构建异构并行通信库,实现高性能设备通信加速库
  • 面向未来低时延、高带宽的计算通信需求进行技术攻关
  • 负责大规模分布式训练中多机计算、通信和训练性能的优化

任职要求

  • 本科及以上学历,计算机、软件工程、通信等相关专业
  • 熟练掌握C/C++,有扎实的编程基础、良好的编程风格和工作习惯
  • 熟悉Python/Shell等脚本语言
  • 具有良好的团队协作精神,责任心强,能够积极主动地完成相关工作
  • 具备并行程序开发经验,了解CUDA、OpenMPI、NCCL等并行程序开发和性能调优经验者优先
  • 掌握计算机系统结构知识和指令流水,有Pipeline优化经验者优先
  • 有集合通信算法开发、网络编程、RDMA编程相关经验者优先
  • 了解Pytorch/Megatron-LM等框架分布式执行方式者优先

福利待遇

  • 15薪薪酬结构
  • 具有竞争力的薪资水平(25-35K)

工作地址

北京海淀区致真大厦D座16层