Job Description

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平台，负责芯片全栈存储架构，解决多线程、多虚拟机、多卡互联场景下内存可见性、硬件隔离、访存带宽瓶颈问题，支撑大模型、自动驾驶、云端虚拟化业务落地。

投递

About XPENG

XPeng is a leading Chinese Smart EV company that designs, develops, manufactures, and markets Smart EVs that appeal to the large and growing base of technology-savvy middle-class consumers. Its mission is to drive Smart EV transformation with technology and data, shaping the mobility experience of the future. In order to optimize its customers’ mobility experience, XPeng develops in-house its full-stack advanced driver-assistance system technology and in-car intelligent operating system, as well as core vehicle systems including powertrain and the electrical/electronic architecture. XPeng is headquartered in Guangzhou, China. In 2021, the Company established its European headquarters in Amsterdam, along with other dedicated offices in Copenhagen, Munich, Oslo, and Stockholm.The Company’s Smart EVs are mainly manufactured at its plant in Zhaoqing and Guangzhou，Guangdong province.

For more information, please visit https://heyxpeng.com.

Industry

Automotive & Mobility

Company Size

1,001-5,000 employees

Headquarters

Guangzhou, CN

Year Founded

2014

Website

xiaopeng.com

Social Media