AI存储架构设计工程师

职位描述

岗位职责： 1. AI多级存储分层架构设计 - 统筹寄存器、Shared Memory、L0/L1/L2缓存、片上全局内存、主机系统内存分层存储架构； - 定义与设计GPGPU MMU、Cache架构； - 定义与设计异步拷贝引擎cp.async、张量访存、批量DMA、MMIO寄存器访问通路，区分volatile/mmio特殊访存语义硬件实现； - 针对大模型训练/推理场景优化带宽、延迟、冲突化解、读写合并、预取策略，平衡算力与存储吞吐。 2. GPU内存一致性模型（MCM）架构定义与RTL约束 - 基于Memory Consistency Model，完成GPGPU片上存储、访存流水线、多级缓存、DMA、IOMMU/SMMU整体一致性架构设计； - 定义Memory Proxy（generic/async/tensormap/fabric）硬件通路隔离、跨Proxy同步、proxy fence屏障实现规范； - 梳理Scope（CTA/Cluster/GPGPU/System）硬件可见域、VMID/SID/PASID地址翻译隔离逻辑； - 负责Release/Acquire、Fence-SC、Barrier、异步访存cp.async等同步原语硬件架构设计。 2. AI加速卡虚拟化I/O存储架构设计 - 设计SR-IOV/MIG硬件虚拟化存储隔离方案，基于Stream ID、VMID实现虚拟机间DMA内存硬件隔离； - 完成SVA共享虚拟地址、PASID子流ID、IOMMU页表翻译、Stage1/Stage2两级地址映射架构； - 解决多虚拟机、多MIG切片并发访存冲突、数据竞争、混合尺寸访存原子性问题，输出虚拟化存储隔离规范。 3. 数据竞争、原子性与同步硬件方案落地 -输出架构spec、架构文档，指导设计与验证团队完成存储单元设计与验证。 4. 前沿架构预研 - 跟踪新一代GPU/AI芯片内存一致性、分布式存储、多卡Fabric、MIG虚拟化、统一地址空间技术； - 针对大模型、多租户云推理、自动驾驶DRIVE场景输出存储架构预研方案与性能评估报告。

职位要求

任职要求： 一、硬性基础要求： 1. 本科及以上，微电子、集成电路、计算机体系结构、电子工程等相关专业；3年及以上GPU/AI芯片/高性能处理器存储架构设计经验； 2. 精通CPU/GPU内存一致性模型、缓存一致性、访存流水线、多级缓存架构，熟悉PTX/ISA内存语义优先； 3. 掌握IOMMU/SMMU、SR-IOV、MIG、VF直通、Stream ID、VMID、PASID虚拟化I/O存储隔离架构； 4. 熟悉硬件同步原语：fence、barrier、atomic RMW、release/acquire、proxy fence硬件实现； 5. 能够独立输出架构Spec、微架构文档； 6. 熟悉Verilog/SystemVerilog，RTL代码，熟悉存储子系统仿真流程。 二、专业深度加分项（核心匹配本岗位）： 1. 熟悉Memory Consistency Model，理解Memory Proxy、Scope、因果序、通信序、一致性； 2. 有GPU异步访存cp.async、张量访存、NVLink多卡Fabric存储架构设计经验； 3. 做过vGPU/MIG虚拟化、多租户硬件隔离、SVA共享虚拟地址相关架构； 4. 熟悉HBM/DDR缓存、片上共享内存、存储通路差异化设计； 6. 熟悉AI大模型训练/推理、自动驾驶DRIVE计算平台存储负载特征。 三、软素质要求： 1. 良好跨团队沟通能力，可对齐编译器、驱动、RTL、验证、后端多团队； 2. 严谨细致，擅长Spec编写； 3. 具备自驱力，能独立完成架构预研、方案对比、性能瓶颈分析。 岗位职责补充（业务场景） 面向通用推理加速卡、车载自动驾驶GPU平台，负责芯片全栈存储架构，解决多线程、多虚拟机、多卡互联场景下内存可见性、硬件隔离、访存带宽瓶颈问题，支撑大模型、自动驾驶、云端虚拟化业务落地。