高性能网络研发工程师

Job description

1、负责小米AI平台高性能网络研发、通信库开发和优化，支撑大模型训练和推理； 2、包括但不限于推训高性能网络的设计、选型和验收；相关驱动和库的问题定位和优化；通信库的开发、优化和维护；推训任务的相关问题定位和排查等。

Qualifications

• 硕士及以上学历，计算机、软件工程等相关专业，具备扎实计算机理论基础和丰富编程经验；
• 熟悉 C++ / Python、操作系统原理和Linux内核原理；
• 熟悉网络基本原理。

加分项

• 熟悉Linux网络驱动，特别是硬件和RDMA驱动开发；
• 熟悉RDMA原理和编程；
• 熟悉常用通信库原理，如NCCL、DeepEP等；
• 熟悉网络原理，如交换机侧负载均衡、拥塞控制等；
• 熟悉CUDA编程。