岗位职责
- 负责小米AI平台高性能网络研发、通信库开发和优化，支撑大模型训练和推理；
- 包括但不限于推训高性能网络的设计、选型和验收；相关驱动和库的问题定位和优化；通信库的开发、优化和维护；推训任务的相关问题定位和排查等。
职位要求
- 硕士及以上学历，计算机、软件工程等相关专业，具备扎实计算机理论基础和丰富编程经验；
- 熟悉 C++ / Python、操作系统原理和Linux内核原理；
- 熟悉网络基本原理。 加分项：
- 熟悉Linux网络驱动，特别是硬件和RDMA驱动开发；
- 熟悉RDMA原理和编程；
- 熟悉常用通信库原理，如NCCL、DeepEP等；
- 熟悉网络原理，如交换机侧负载均衡、拥塞控制等；
- 熟悉CUDA编程。
