职位描述
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等集合通信性能优化经验。
