分布式训练研发工程师
岗位摘要
基于MLU构建异构并行通信库,实现高性能设备通信加速库;面向未来低时延、高带宽的计算通信需求进行技术攻关;负责大规模分布式训练中多机计算、通信和训练性能的优化;本科及以上学历,计算机、软件工程、通信等相关专业
登录后保留你的岗位线索
邮箱登录后可继续查看完整职责、任职要求和原岗位入口,并使用收藏与浏览记录沉淀适合自己的机会。
岗位职责
- 基于MLU构建异构并行通信库,实现高性能设备通信加速库
- 面向未来低时延、高带宽的计算通信需求进行技术攻关
- 负责大规模分布式训练中多机计算、通信和训练性能的优化
任职要求
- 本科及以上学历,计算机、软件工程、通信等相关专业
- 熟练掌握C/C++,有扎实的编程基础、良好的编程风格和工作习惯
- 熟悉Python/Shell等脚本语言
- 具有良好的团队协作精神,责任心强,能够积极主动地完成相关工作
- 具备并行程序开发经验,了解CUDA、OpenMPI、NCCL等并行程序开发和性能调优经验者优先
- 掌握计算机系统结构知识和指令流水,有Pipeline优化经验者优先
- 有集合通信算法开发、网络编程、RDMA编程相关经验者优先
- 了解Pytorch/Megatron-LM等框架分布式执行方式者优先
福利待遇
- 15薪薪酬结构
- 具有竞争力的薪资水平(25-35K)
工作地址
北京海淀区致真大厦D座16层
登录后查看完整岗位详情
使用邮箱验证码登录后,可查看完整职责、任职要求、原岗位入口,并继续使用收藏和浏览记录。列表摘要保持公开,完整详情用于保护数据质量和降低恶意抓取。