logo of alibaba

阿里巴巴AI推理平台-大模型异构芯片适配与推理优化工程师-杭州/北京

社招全职2年以上地点:北京 | 杭州状态:招聘

任职要求


1. 负责 Qwen、DeepSeek、GLM、Kimi等主流模型在 TPU、NPU 等各类AI 加速芯片上的适配、验证和性能优化。
2. 分析模型结构与硬件特性,完成 attention、KV cache、MoE、多模态模块、DiT/扩散模型等关键链路的推理打通与优化。
3. 基于 vLLM、SGLang 或自研推理框架,推进模型部署、算子替换、图编译、ke…
登录查看完整任职要求
微信扫码,1秒登录

工作职责


负责主流大模型在多种 AI 加速芯片和推理后端上的适配、优化与工程落地,覆盖 Qwen、DeepSeek、GLM、Kimi等文本、多模态与图像生成模型,目标平台包括 TPU、NPU 等各类 AI 加速芯片。该岗位需要深入理解模型结构、推理框架、算子/kernel、编译栈与分布式推理机制,推动模型在不同硬件平台上稳定、高效、可规模化运行。
包括英文材料
缓存+
vLLM+
还有更多 •••
相关职位

logo of aligenie
社招4年以上

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 服务的高可用与容错能力。

更新于 2026-06-16北京|杭州
logo of aligenie
社招4年以上

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 利用率和请求成功率的影响,持续优化系统在高并发、多租户、长上下文场景下的服务质量。

更新于 2026-06-16北京|杭州
logo of alibaba
社招4年以上

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 服务的高可用与容错能力。

更新于 2026-06-16北京|杭州
logo of alibaba
社招4年以上

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 利用率和请求成功率的影响,持续优化系统在高并发、多租户、长上下文场景下的服务质量。

更新于 2026-07-28北京|杭州