AI芯片互联软件工程师-Data

职位描述

1、参与AI芯片多卡互联方案的设计与开发，开发和优化多卡互联的软件驱动程序，实现多卡之间的协同工作；

2、参与多卡互联系统的验证与调试，及时发现并解决在测试过程中出现的各种问题；

3、参与AI芯片互联相关软件栈开发，包括Kernel Mode Driver/User Mode Driver/SDK/Tools等；

4、跟踪行业最新技术动态，为公司的AI芯片多卡互联技术发展提供前瞻性的建议和技术储备。

任职要求

1、计算机/电子信息工程相关专业本科及以上学历；

2、熟练掌握至少一种编程语言，如C、C++等，熟练掌握Linux系统编程；有Linux内核驱动开发经验，熟练Linux操作系统下的驱动开发流程；

3、至少2年以上AI芯片/GPGPU芯片/RDMA网卡芯片或者相关领域的工作经验；

4、对多卡互联技术有深入的理解，包括不限于NVLink、SUE、RDMA或者其他高速互联接口协议；掌握RDMA网卡的驱动开发，熟悉TCP/IP协议，熟悉RoCE V2协议；

5、熟悉NCCL等集合通信库的原理以及使用；

6、具有良好的团队合作精神和沟通能力，能够与跨部门团队（如硬件团队、算法团队等）紧密协作。

加分项：

1、理解LLM网络模型结构，熟悉模型部署、分析和优化；

2、擅长跨团队沟通交流协作，有项目或团队管理经验。