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