岗位描述
岗位职责
1. 负责研发 AI 芯片的通信算子库:开发AllReduce、AllGather、ReduceScatter、Broadcast等集合通信原语。
2. 开发通信协议与调度策略模块,持续优化数据传输与通信性能。
3. 基于PCIe、RDMA等互联拓扑优化通信算法,提升带宽并降低时延。分析和优化 AI 集群中的通信算子性能,提升整体系统效率;
4. 开发通信与计算Overlap及通信域管理,支持TP、PP、DP等训练场景。
5. 开发PyTorch Distributed ProcessGroup Backend,支持主流AI框架集成。
6. 编写单元、性能及自动化测试,定位修复通信问题,配合联调优化。
岗位要求
1. 计算机、软件、通信等相关专业本科及以上学历,硕士及以上学历优先。
2. 熟练掌握C/C++,熟悉Linux开发环境及Git、CMake等工具。
3. 熟悉多线程、锁、原子操作、异步编程及常见数据结构与算法。
4. 了解集合通信原理,理解Ring、Tree等通信算法。
加分项
a, 有NCCL、HCCL、RCCL、MPI等开源集合通信库开发经验。
b, 有AI芯片通信软件开发及RDMA、InfiniBand、RoCE开发经验。
c, 有AllReduce、AllGather、All-to-All等集合通信性能优化经验。