岗位职责
- 参与AI芯片Profiling方案设计,负责Profiling模块的验证以及相关驱动和工具的开发;
- 参与AI芯片任务调度器驱动方案设计和开发;
- 参与AI芯片任务调度器Firmware方案设计和开发;
- 跟踪行业最新技术动态,为公司的AI芯片多卡互联技术发展提供前瞻性的建议和技术储备。
职位要求
- 计算机/自动化相关专业本科及以上学历;
- 熟练掌握至少一种编程语言,如C、C++等,熟练掌握Linux系统编程,有Linux内核驱动开发经验,熟练Linux操作系统下的驱动开发流程;
- 至少2年以上AI芯片/GPGPU芯片或者相关领域的工作经验;
- 熟悉AI芯片任务调度器相关软件开发;
- 熟悉CUDA Runtime/Driver API/Cupti/Nsys/Nsight等,熟悉CUDA Toolkit软件;
- 具有良好的团队合作精神和沟通能力,能够与跨部门团队(如硬件团队、算法团队等)紧密协作,共同推进项目进展。 加分项:
- 理解LLM网络模型结构,熟悉模型部署、分析和优化;
- 擅长跨团队沟通交流协作,有项目或团队管理经验。