互联软件工程师(AI芯片) - 芯片研发
ByteDance · Beijing; Shanghai; Hangzhou; Shenzhen; Xi'an ·
团队介绍:AI芯片团队深耕人工智能芯片研发领域,产品覆盖云端与终端场景,致力于提供高效率、低成本的AI芯片及 SoC 解决方案,满足公司内外AI及相关业务的算力需求。团队自主设计了多款AI芯片,涵盖推理、训练场景,具有高性能、高精度、高灵活性、编译器友好、低成本等特点。核心AI加速核与AI软件栈均为自主研发,具备完整知识产权,能够支撑面向不同模型、不同部署场景的持续演进与优化。
1、参与AI芯片多卡互联方案的设计与开发,开发和优化多卡互联的软件驱动程序,实现多卡之间的协同工作;
2、参与多卡互联系统的验证与调试,及时发现并解决在测试过程中出现的各种问题;
3、参与AI芯片互联相关软件栈开发,包括Kernel Mode Driver/User Mode Driver/SDK/Tools等;
4、跟踪行业最新技术动态,为公司的AI芯片多卡互联技术发展提供前瞻性的建议和技术储备。
1、计算机/电子信息工程相关专业本科及以上学历;
2、熟练掌握至少一种编程语言,如C、C++等,熟练掌握Linux系统编程,有Linux内核驱动开发经验,熟练Linux操作系统下的驱动开发流程,掌握RDMA网卡的驱动开发,熟悉TCP/IP协议,熟悉RoCE V2协议;
3、至少2年以上AI芯片/GPGPU芯片/RDMA网卡芯片或者相关领域的工作经验;
4、对多卡互联技术有深入的理解,包括不限于NVLink、SUE、RDMA或者其他高速互联接口协议;
5、熟悉NCCL等集合通信库的原理以及使用;
6、具有良好的团队合作精神和沟通能力,能够与跨部门团队(如硬件团队、算法团队等)紧密协作。
加分项:
1、理解LLM网络模型结构,熟悉模型部署、分析和优化;
2、擅长跨团队沟通交流协作,有项目或团队管理经验。