返回岗位列表

xAI

工程师

地点:美国 薪资:$180,000-$440,000 日期:2026-01-29

岗位摘要

开发和调优基于RDMA的通信系统,利用NVIDIA GPU和Mellanox网卡(InfiniBand, RoCE)实现节点间的超高速数据传输;实施和优化GPUDirect RDMA,以实现GPU和网络接口之间的直接内存访问,最小化CPU开销

岗位职责

  • 开发和调优基于RDMA的通信系统,利用NVIDIA GPU和Mellanox网卡(InfiniBand, RoCE)实现节点间的超高速数据传输
  • 实施和优化GPUDirect RDMA,以实现GPU和网络接口之间的直接内存访问,最小化CPU开销
  • 将RDMA解决方案与基于Kubernetes的工作负载集成,确保跨分布式计算和存储系统的无缝操作
  • 与AI研究人员和基础设施团队合作,使用NCCL和MPI加速数据管道和集体通信
  • 排查并解决高吞吐量、低延迟网络环境中的性能瓶颈

任职要求

  • 在高性能计算或AI超级计算环境中,具备NVIDIA RDMA技术(如GPUDirect RDMA, RoCE, InfiniBand)的实践经验
  • 精通使用Rust、C或C++进行底层网络和系统优化编程
  • 具备使用MPI、NCCL或类似框架优化GPU加速分布式系统的经验
  • 了解Kubernetes网络,并有将RDMA集成到容器化环境中的经验
  • 加分项:具备AI/ML训练工作流程及其网络需求(如大规模参数同步)的背景

福利待遇

  • 全面的医疗、视力和牙科保险
  • 401(k)退休计划
  • 短期和长期残疾保险
  • 各种其他折扣和福利