字节跳动AI芯片系统软件架构师/工程师
任职要求
1、计算机/自动化相关专业本科及以上学历; 2、熟练掌握至少一种编程语言,如C、C++等,熟练掌握Linux系统编程; 3、有Linux内核驱动开发经验,熟练Linux操作系统下的驱动开发流程; 4、至少2年以上AI芯片以及GPGPU芯片或者相关领域的工作经验; 5、熟悉AI芯片或者GPGPU架构和工作原理,对多卡互联技术有深入的理解,包括但不限于PCIe、NVLink、RoCEV2等高速互联接口协议;熟悉CUDA Runtime/Driver API/NCCL/CUPTI等,熟悉CUDA Toolkit软件栈; 6、具有良好的团队合作精神和沟通能力,能够与跨部门团队(如硬件团队、算法团队等)紧密协作,共同推进项目进展;具备较强的问题解决能力和学习能力,能够快速解决工作中遇到的技术难题,并不断学习和掌握新的技术和知识。 加分项: 1、理解LLM网络模型结构,熟悉模型部署、分析和优化; 2、擅长跨团队沟通交流协作,有项目或团队管理经验。
工作职责
1、参与AI芯片多卡互联方案的设计与开发,根据公司产品需求,制定高效、稳定的多卡互联架构; 2、参与开发和优化多卡互联的软件驱动程序,实现多卡之间的协同工作,提高系统的整体性能和稳定性; 3、参与多卡互联系统的验证与调试,及时发现并解决在测试过程中出现的各种问题; 4、参与AI芯片Profiling方案设计,负责Profiling模块的验证以及相关驱动和工具的开发; 5、参与AI芯片任务调度器驱动方案设计和开发,负责任务调度Firmware的开发; 6、跟踪行业最新技术动态,为公司的AI芯片多卡互联技术发展提供前瞻性的建议和技术储备。
1、参与AI芯片多卡互联方案的设计与开发,根据公司产品需求,制定高效、稳定的多卡互联架构; 2、参与开发和优化多卡互联的软件驱动程序,实现多卡之间的协同工作,提高系统的整体性能和稳定性; 3、参与多卡互联系统的验证与调试,及时发现并解决在测试过程中出现的各种问题; 4、参与AI芯片Profiling方案设计,负责Profiling模块的验证以及相关驱动和工具的开发; 5、参与AI芯片任务调度器驱动方案设计和开发,负责任务调度Firmware的开发; 6、跟踪行业最新技术动态,为公司的AI芯片多卡互联技术发展提供前瞻性的建议和技术储备。
1、参与AI芯片多卡互联方案的设计与开发,根据公司产品需求,制定高效、稳定的多卡互联架构; 2、参与开发和优化多卡互联的软件驱动程序,实现多卡之间的协同工作,提高系统的整体性能和稳定性; 3、参与多卡互联系统的验证与调试,及时发现并解决在测试过程中出现的各种问题; 4、参与AI芯片Profiling方案设计,负责Profiling模块的验证以及相关驱动和工具的开发; 5、参与AI芯片任务调度器驱动方案设计和开发,负责任务调度Firmware的开发; 6、跟踪行业最新技术动态,为公司的AI芯片多卡互联技术发展提供前瞻性的建议和技术储备。
1、参与AI芯片多卡互联方案的设计与开发,根据公司产品需求,制定高效、稳定的多卡互联架构; 2、参与开发和优化多卡互联的软件驱动程序,实现多卡之间的协同工作,提高系统的整体性能和稳定性; 3、参与多卡互联系统的验证与调试,及时发现并解决在测试过程中出现的各种问题; 4、参与AI芯片Profiling方案设计,负责Profiling模块的验证以及相关驱动和工具的开发; 5、参与AI芯片任务调度器驱动方案设计和开发,负责任务调度Firmware的开发; 6、跟踪行业最新技术动态,为公司的AI芯片多卡互联技术发展提供前瞻性的建议和技术储备。