GPGPU软件栈工程师/专家(J10785)

职位描述

岗位职责：​

1. Runtime系统设计与开发​：参与设计、开发GPGPU软件栈，包括Runtime和用户态驱动（UMD）、内核态（KMD）等核心组件。

2. 新特性实现与优化​：依据AI与通用计算的应用需求，开发实现GPGPU驱动新特性；针对GPGPU硬件新特性，开发相应的软件支持。

3. 性能分析与调优​：对关键应用进行性能分析和性能优化，解决多级流水线、低开销上下文切换、异构同步等系统级问题。

4. 跨团队协作​：与硬件团队合作定义计算单元架构，驱动ISA（指令集架构）设计；与算法团队合作优化计算库（如BLAS、FFT）的硬件加速策略。

5. 工具链开发​：开发Profiling工具链，定位内核执行热点，提出架构级改进方案。

任职要求

任职要求：​

1. 学历与经验​：

- 计算机科学、电子工程、数学等相关专业本科及以上学历。

- 8年以上系统级软件开发经验，有GPU或GPGPU UMD驱动开发经验者优先。

2. 技术能力​：

- 编程语言​：精通C/C++编程语言，具备坚实的开发调试经验。

- 异构计算​：熟悉至少一种通用计算API，如CUDA、HIP、OpenCL等，并深入理解其Runtime实现原理（如CUDA Driver API、HIP Runtime、OpenCL任务调度）。

- 系统架构​：熟悉操作系统架构（Linux/Unix），精通Linux内核模块、DMA、PCIe通信等底层开发技术。

- 硬件知识​：对现代GPGPU架构（如NVIDIA Ampere/AMD CDNA）有深入研究，熟悉SIMT执行模型、层级内存体系、硬件调度队列等核心机制。

3. 加分项​：

- 有芯片Bringup项目经验者尤佳。

- 参与过国际开源异构计算项目（如PyTorch/XLA、TVM、Vulkan计算管线）并贡献核心代码。

- 有GPGPU/FPGA/ASIC等芯片厂商底层工具链开发经验，熟悉指令集模拟器（ISS）或性能建模工具。

优先考虑条件：​

- 有AI相关研发经验，熟悉CUDA编程，熟悉AI框架（如TensorFlow、PyTorch），对一种或多种网络模型结构有深入理解。

- 具备跨团队沟通协作能力，有项目或团队管理经验者优先