Ivyea Jobs 10686 roles · 174 companies · 刚刚
云天励飞 · AI 芯片

gpgpu runtime & driver开发工程师

深圳 社招 · 全职 工程开发

职位描述

1、Runtime系统设计与开发:参与设计、开发GPGPU软件栈,包括Runtime和用户态驱动(UMD)、内核态(KMD)等核心组件。 2、新特性实现与优化:依据AI与通用计算的应用需求,开发实现GPGPU驱动新特性;针对GPGPU硬件新特性,开发相应的软件支持。 3、性能分析与调优:对关键应用进行性能分析和性能优化,解决多级流水线、低开销上下文切换、异构同步等系统级问题。 4、跨团队协作:与硬件团队合作定义计算单元架构,驱动ISA(指令集架构)设计;与算法团队合作优化计算库(如BLAS、FFT)的硬件加速策略。 5、工具链开发:开发Profiling工具链,定位内核执行热点,提出架构级改进方案。

任职要求

1、学历与经验: a、计算机科学、电子工程、数学等相关专业本科及以上学历。 b、5年以上系统级软件开发经验,有GPU或GPGPU UMD驱动开发经验者优先。 2、技术能力: a、编程语言:精通C/C++编程语言,具备坚实的开发调试经验。 b、异构计算:熟悉至少一种通用计算API,如CUDA、HIP、OpenCL等,并深入理解其Runtime实现原理(如CUDA Driver API、HIP Runtime、OpenCL任务调度)。 c、系统架构:熟悉操作系统架构(Linux/Unix),精通Linux内核模块、DMA、PCIe通信等底层开发技术。 d、硬件知识:对现代GPGPU架构(如NVIDIA Ampere/AMD CDNA)有深入研究,熟悉SIMT执行模型、层级内存体系、硬件调度队列等核心机制。 3、加分项: a、有芯片Bringup项目经验者尤佳。 b、参与过国际开源异构计算项目(如PyTorch/XLA、TVM、Vulkan计算管线)并贡献核心代码。 c、有GPGPU/FPGA/ASIC等芯片厂商底层工具链开发经验,熟悉指令集模拟器(ISS)或性能建模工具。 优先考虑条件: 有AI相关研发经验,熟悉CUDA编程,熟悉AI框架(如TensorFlow、PyTorch),对一种或多种网络模型结构有深入理解。 具备跨团队沟通协作能力,有项目或团队管理经验者优先
云天励飞去官方页面投递 →