阿里巴巴大模型推理KV Cache系统:从存储架构到软硬协同的全栈演进-阿里星
任职要求
1. 博士在读或即将毕业,计算机科学、数据库、人工智能、机器学习等相关专业背景; 2. 在大模型推理、数据库和存储等领域有深入研究和系统实践经验。熟悉分布式系统原理,熟悉各类存储协议和硬件特性; 3. 具备扎实的计算机基础,具备良好的编程能力和编程规范,熟悉 C/C++/Python/Go/Rust 至少一门语言; 4. 学习能力强,自驱、专注、善于拆解复杂问题并推动工程落地,愿意思考问题本质,具备质疑精神;具备良好的学术研究与工程化落地能力。 加分项:…
工作职责
大模型推理的成本与效率是行业核心挑战,而 KV Cache 管理是突破瓶颈的关键。本岗位将主导构建下一代 LLM 推理存储架构,融合算法优化、系统调度与硬件感知能力,打造高弹性、低成本、高可用的推理基础设施,支撑公司核心大模型业务的规模化落地。 1. 负责设计和实现面向大规模推理的存储系统,深度融合推理引擎,构建从显存、本地内存至分布式冷存储的全局多级存储池; 2. 基于特征识别实现数据的智能调度与分布管理,推动计算存储分离、弹性资源池等核心能力落地,建立高扩展、高可靠的推理存储系统底座; 3. 定义存储与推理引擎间的统一数据访问抽象层,通过零侵入式接口设计实现计算存储解耦,为多级缓存、预取调度等高级策略提供标准化支撑; 4. 建立 KV Cache 资源评估与治理体系,通过数据驱动的方式识别优化空间,在保障服务品质的前提下持续降低推理成本、提升资源 ROI; 5. 联合上下游团队,深入 KV Cache 全栈优化和技术攻关,建立端到端 KV Cache 系统; 6. 跟踪学术界与工业界 KV Cache 最新进展,推动量化、压缩、调度等创新技术在大规模生产环境中的落地。

1. KVCache 核心系统研发,负责 LLM 推理场景下 KVCache 的架构设计与工程实现,优化 KVCache 的内存管理、显存分配与生命周期调度策略,研究并实现 Prefix Cache、Radix Tree Cache 等高效缓存复用机制,提升 Cache Hit Rate。 2. 设计跨机、跨节点的分布式 KVCache 共享与迁移方案,实现 KVCache 在 GPU HBM / CPU DRAM / NVMe SSD 多级存储间的高效卸载(Offload)与加载,针对长上下文、多轮对话场景优化 KVCache 的存储与传输效率。 3. 推理性能优化,结合 Continuous Batching、PagedAttention、Chunked Prefill 等机制,协同优化 KVCache 调度策略,分析推理链路中 KVCache 相关的性能瓶颈,进行端到端性能优化,针对主流推理框架(vLLM / SGLang / TensorRT-LLM / RTP-LLM)进行 KVCache 模块的深度优化与定制。 4. 系统可靠性与可观测性建设,建设 KVCache 命中率、内存占用、调度延迟等核心指标的监控体系,保障大规模集群下 KVCache 服务的高可用与容错能力。

1. 负责 LLM 推理场景下 KVCache 管控系统的架构设计与工程实现,建设面向大规模 GPU 集群的 KVCache 资源管理、容量规划、配额控制、生命周期管理与调度编排能力,提升 KVCache 资源利用率与系统稳定性。 2. 设计 KVCache 全局管控与调度策略,结合请求特征、模型类型、上下文长度、Prefix 复用关系、租户优先级与集群资源状态,实现 KVCache 的准入控制、淘汰策略、热点识别、跨实例复用、跨节点迁移与负载均衡。 3. 建设 KVCache 多级存储管控能力,统一管理 GPU HBM / CPU DRAM / NVMe SSD / 远端存储中的 Cache 资源,设计 Cache 分层、回收、预热、预取、降级与故障恢复机制,支撑长上下文、多轮对话、Agent 工作流等复杂推理场景。 4. 负责 KVCache 管控系统与主流推理引擎及调度系统的集成,围绕 vLLM / SGLang / TensorRT-LLM / RTP-LLM 等框架,抽象统一的 KVCache 元数据、状态同步、资源上报与控制接口,支撑异构推理后端的统一纳管。 5. 建设 KVCache 可观测性与稳定性体系,设计 Cache Hit Rate、复用收益、内存水位、碎片率、迁移延迟、Offload 延迟、淘汰次数、请求级 Cache 轨迹等核心指标,支持容量评估、异常诊断、策略调优与线上问题定位。 6. 参与大规模推理集群的端到端性能优化,分析 KVCache 管控策略对 TTFT、TPOT、吞吐、显存利用率、GPU 利用率和请求成功率的影响,持续优化系统在高并发、多租户、长上下文场景下的服务质量。
1. KVCache 核心系统研发,负责 LLM 推理场景下 KVCache 的架构设计与工程实现,优化 KVCache 的内存管理、显存分配与生命周期调度策略,研究并实现 Prefix Cache、Radix Tree Cache 等高效缓存复用机制,提升 Cache Hit Rate。 2. 设计跨机、跨节点的分布式 KVCache 共享与迁移方案,实现 KVCache 在 GPU HBM / CPU DRAM / NVMe SSD 多级存储间的高效卸载(Offload)与加载,针对长上下文、多轮对话场景优化 KVCache 的存储与传输效率。 3. 推理性能优化,结合 Continuous Batching、PagedAttention、Chunked Prefill 等机制,协同优化 KVCache 调度策略,分析推理链路中 KVCache 相关的性能瓶颈,进行端到端性能优化,针对主流推理框架(vLLM / SGLang / TensorRT-LLM / RTP-LLM)进行 KVCache 模块的深度优化与定制。 4. 系统可靠性与可观测性建设,建设 KVCache 命中率、内存占用、调度延迟等核心指标的监控体系,保障大规模集群下 KVCache 服务的高可用与容错能力。
1. 负责 LLM 推理场景下 KVCache 管控系统的架构设计与工程实现,建设面向大规模 GPU 集群的 KVCache 资源管理、容量规划、配额控制、生命周期管理与调度编排能力,提升 KVCache 资源利用率与系统稳定性。 2. 设计 KVCache 全局管控与调度策略,结合请求特征、模型类型、上下文长度、Prefix 复用关系、租户优先级与集群资源状态,实现 KVCache 的准入控制、淘汰策略、热点识别、跨实例复用、跨节点迁移与负载均衡。 3. 建设 KVCache 多级存储管控能力,统一管理 GPU HBM / CPU DRAM / NVMe SSD / 远端存储中的 Cache 资源,设计 Cache 分层、回收、预热、预取、降级与故障恢复机制,支撑长上下文、多轮对话、Agent 工作流等复杂推理场景。 4. 负责 KVCache 管控系统与主流推理引擎及调度系统的集成,围绕 vLLM / SGLang / TensorRT-LLM / RTP-LLM 等框架,抽象统一的 KVCache 元数据、状态同步、资源上报与控制接口,支撑异构推理后端的统一纳管。 5. 建设 KVCache 可观测性与稳定性体系,设计 Cache Hit Rate、复用收益、内存水位、碎片率、迁移延迟、Offload 延迟、淘汰次数、请求级 Cache 轨迹等核心指标,支持容量评估、异常诊断、策略调优与线上问题定位。 6. 参与大规模推理集群的端到端性能优化,分析 KVCache 管控策略对 TTFT、TPOT、吞吐、显存利用率、GPU 利用率和请求成功率的影响,持续优化系统在高并发、多租户、长上下文场景下的服务质量。