GPU计算驱动工程师(J10966)

职位描述

工作职责：

1. 基于通用计算发展趋势，定义摩尔线程计算编程模型及接口（MUSA），设计和开发相应驱动及运行时程序；

2. 与图形驱动团队协作开发底层公共层；

3. 与开发者工具链及编译器等团队协作开发性能分析工具、调试器、Sanitizer工具、代码自动移植工具等；

4. 为新引入的IP设计开发驱动程序；

5. 协同定位解决上层应用运行时问题，如大规模分布式训练、推理、超算系统的问题；

6. 协同架构、编译器、加速库及开发者工具等团队，联合优化各业务场景端到端性能；

7. 深度参与下一代芯片架构软硬协同设计及优化、性能仿真；

任职要求

任职要求：

1. 精通C/C++编程，具有良好的编程习惯，熟悉常见数据结构和算法，熟悉Linux/Windows系统编程接口；

2. 熟悉计算机体系结构，操作系统原理，编译原理；

3. 熟悉异构计算体系结构，熟悉以下至少一种异构计算API：CUDA、HIP、HSA、OpenCL、Vulkan；

4. 具有AI训练、RL、推理场景硬件协同优化、框架及驱动联合优化经验者优先；

5. 具有较强的定位和解决问题的能力；

6. 具有驱动开发经验、shader编程和优化经验者优先；

加分项：

1. 具有AI应用开发、科学计算应用开发、驱动开发经验、GPU应用开发经验，了解GPU架构；

2. 在国际顶尖会议或期刊（CVPR、ICCV、ECCV、3DV、SIGGRAPH、TOG、TPAMI、ACL、NAACL、EMNLP、AAAI等）上发表过学术论文；

3. 在有影响的计算机科学相关的竞赛中获奖，如ASC、ACM、NOI等；