JobConnect

高性能AI网络与集合通信技术工程师

  • NIO
  • Shanghai, Shenzhen +2 more, China
  • CNY 400,000 – CNY 800,000

上海、深圳、杭州、合肥社招全职数字技术 - 芯片研发本科及以上1-3 年

职位描述

  1. 结合自研芯片片上互联、多芯片拓扑架构、高速硬件通信特性,参与自研通信库软件研发,涵盖 Ring/Tree 结构通信、计算通信流水重叠、动态负载均衡、流量控制等技术,有效降低通信延迟、提升带宽利用率。
  2. 深度对接大模型分布式训推、Agent等业务场景,搭建完善的多芯片 / 多机通信性能基准,结合拓扑、协议、算法与模型,跨层次探索下一代 ai 网络架构。
  3. 跟踪 NCCL、MPI、NIXL、DeepEP、MoonEP 等相关行业主流通信技术与前沿技术趋势,开展技术预研、原型验证与技术创新,沉淀专利、技术成果,推动自研集合通信软件栈产品落地。

职位要求

  1. 学历专业:本科、硕士及以上学历。
  2. 工作经验:1-5 年及以上高性能网络、分布式通信库、多机 / 多卡互联优化相关工作经验,有 AI 芯片集群通信栈研发落地经验者优先。
  3. 技术能力:熟悉 C/C++、Linux 系统编程,熟练使用 Python,具备扎实的软件开发、调试与性能调优能力。熟练掌握AI coding高效开发经验者优先。
  4. 熟悉高速网络技术原理,了解 RDMA、RoCE、IB、芯片间高速互联架构,具备通信算法建模、复杂度分析、仿真落地的实战能力。有 NCCL、MPI、NIXL、DeepEP、MoonEP 等开源通信库二次开发、定制优化、架构改造实战经验,熟悉 ibverbs、RDMA 编程者优先。
  5. 具备 AI 芯片多卡 / 多机分布式训练、推理场景通信优化经验,熟悉 Megatron-LM、DeepSpeed、vLLM、sglang 等任一分布式框架经验者优先;有国产算力芯片互联通信栈自研项目经验者优先。

投递

Skills

  • C/C++
  • Linux
  • Python
  • RDMA
  • NCCL
  • MPI

Related jobs

NIOApply for this job