AI芯片计算软件架构师/工程师-Data

职位描述

1、参与AI芯片Profiling方案设计，负责Profiling模块的验证以及相关驱动和工具的开发；

2、参与AI芯片任务调度器驱动方案设计和开发；

3、参与AI芯片任务调度器Firmware方案设计和开发；

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

任职要求

1、计算机/自动化相关专业本科及以上学历；

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

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

4、熟悉AI芯片任务调度器相关软件开发；

5、熟悉CUDA Runtime/Driver API/Cupti/Nsys/Nsight等，熟悉CUDA Toolkit软件；

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

加分项：

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

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