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

技术开发、技术推广、技术转让、技术咨询(中介除外)、技术服务;技术进出口、货物进出口(国营贸易管理货物除外);计算机系统服务;基础软件服务;应用软件服务;软件开发;软件咨询(中介除外);销售计算机、软件及辅助设备、自行开发后的产品。(企业依法自主选择经营项目,开展经营活动;依法须经批准的项目,经相关部门批准后依批准的内容开展经营活动;不得从事本市产业政策禁止和限制类项目的经营活动。)