岗位职责
- 参与AI芯片多卡互联方案的设计与开发,开发和优化多卡互联的软件驱动程序,实现多卡之间的协同工作;
- 参与多卡互联系统的验证与调试,及时发现并解决在测试过程中出现的各种问题;
- 参与AI芯片互联相关软件栈开发,包括Kernel Mode Driver/User Mode Driver/SDK/Tools等;
- 跟踪行业最新技术动态,为公司的AI芯片多卡互联技术发展提供前瞻性的建议和技术储备。
职位要求
- 计算机/电子信息工程相关专业本科及以上学历;
- 熟练掌握至少一种编程语言(如C、C++等)及Linux系统编程,具备Linux内核驱动开发经验,熟悉驱动开发流程;
- 至少2年以上AI芯片/GPGPU芯片/RDMA网卡芯片或者相关领域的工作经验;
- 对多卡互联技术有深入理解(包括但不限于NVLink、SUE、RDMA等高速互联接口协议),掌握RDMA网卡驱动开发,熟悉TCP/IP协议及RoCE V2协议;
- 熟悉NCCL等集合通信库的原理以及使用;
- 具有良好的团队合作精神和沟通能力,能够与跨部门团队(如硬件团队、算法团队等)紧密协作。 加分项:
- 理解LLM网络模型结构,熟悉模型部署、分析和优化;
- 擅长跨团队沟通交流协作,有项目或团队管理经验。