Ivyea Jobs 5947 roles · 66 companies · 刚刚
天数智芯 · AI 芯片

GPU通信库工程师

南京 / 西安 / 上海 社招 · 全职 技术类 AI Infra / 训练系统

职位描述

岗位职责: 1. 研发GPGPU通信库; 2. 定位和解决应用中的分布式通信问题; 3. 分析优化分布式计算中的单机内/多机间通信性能。

任职要求

1. 熟悉C/C++编程; 2. 熟悉分布式常用的集合通信操作,了解常用通信库,如NCCL、NVSHMEM、DeepEP等; 3. 熟悉网络通信、RDMA技术,了解ibverbs编程接口; 4. 熟悉分布式训练推理框架,如PyTorch、Megatron-LM、vLLM; 5. 了解GPU体系架构和CUDA编程者优先; 6. 有类NV通信库开发经验者优先; 7. 有大规模集群、超节点通信调优经验者优先。 立即投递