岗位职责
- 负责自研 AI 芯片计算 Runtime 的核心研发,涵盖内存管理、流/事件调度、kernel 启动路径、多设备协同等模块; - 参与自研 AI 芯片硅前验证与 bring-up 全流程:硅前基于 FPGA / Emulator 等平台先行开发并验证 Runtime 软件栈;支持硅后软件栈,与硬件、验证团队联合调试,推动芯片快速进入可交付状态; - 与FW、KMD、编译工具链及 PyTorch / vLLM 等上层框架联调,打通端到端功能,负责疑难问题定位与根因分析; - 负责性能优化:建设 profiling 与 benchmark 体系,围绕 kernel 启动开销、调度与同步路径、内存传输效率、多卡协同等维度持续优化 Runtime 的延迟与吞吐,对标业界领先水平; - 参与技术方案设计与评审,沉淀设计文档,持续提升团队代码质量与工程效率。
职位要求
- 计算机相关专业,3 年以上 C++ 开发经验,精通 C++ 语言与工程实践,能编写高性能、线程安全的工业级代码;
- 具备扎实的计算机体系结构与操作系统基础,熟悉 Cache、NUMA、内存一致性等机制,熟练掌握常用算法与数据结构;
- 熟悉 CUDA、ROCm HIP、OpenMP、OpenCL 等至少一种并行编程模型,理解其底层调度、同步与内存模型,有实际的并行优化落地经验;
- 对代码质量与性能有很高的标准,具备技术热情,能够在新领域快速开拓。 加分项 1. 有芯片硅前验证或 bring-up 经验,熟悉在 FPGA / Emulator / Zebu / HAPS 等平台上的软件开发与软硬件协同调试;
- 有并行计算 Runtime 研发经验,如 AI 芯片 Runtime、CUDA Runtime、ROCm HIP、oneAPI 等;
- 熟悉 PCIe、mailbox / doorbell 等软硬件交互机制。