2027校招-通信库开发工程师(北京/上海/成都)
职位描述
"负责 GPU 集合通信库开发与优化,服务大模型分布式训练。
主要工作
维护 / 二次开发 NCCL,调优 All‑Reduce、All‑Gather 等集合通信算子;
RDMA、IB、以太网多链路性能调优,解决多机多卡通信瓶颈;
定位分布式训练中的通信报错、超时、性能低下问题;
对接 PyTorch/TensorFlow
协作对象:训练、推理、GPU 驱动、集群运维。"任职要求
精通 C/C++、Linux 开发调试,掌握计算机网络、体系结构,了解 GPU 编程、CUDA,熟悉集合通信、RDMA 优先。懂 NCCL 源码、分布式训练框架者加分,侧重性能优化与问题排查能力。