Skip to content
简体中文

切换主题

// AI 新闻 · 2026-11-11 · PyTorch Blog

Meta在TPU v7上优化Muse Spark Open模型推理

Meta在其单个TPU v7主机(含四芯片)上部署了816B参数的Muse Spark Open混合专家模型,采用SGLang与TorchTPU后端运行。核心为一个megakernel,将62层解码或验证步骤整合为单一Pallas程序,并通过流水线方式处理权重加载、KV读取及跨设备传输。其余优化包括更高效的解码流程、每芯片更高的KV缓存容量以及更快的预填充阶段。

原文发布 2026-11-11 00:45(北京时间)来源 PyTorch Blog自动整理 · 已核对原文

01摘要

该方案在单请求输出token速度上实现显著提升,尤其在并发数为1至2时效果明显;验证步骤耗时接近仅从HBM加载权重的时间。基于SemiAnalysis的AgentX基准测试,在1M上下文场景下,每芯片吞吐量从6,101提升至25,906 tokens/s,p90响应时间控制在47秒以内,且所有TPU节点均满足该延迟要求。

由开源模型根据原文自动整理,经数字/专名逐项核对(必须在原文中出现),以原文为准。

02要点

03原文出处

Optimizing Muse Spark Open on TPU v7 with SGLang-TorchTPU

PyTorch Blog · 发布于 2026-11-11 00:45(北京时间) · 上方为原标题

05相关新闻

出自智图 AI 新闻:从 1100 多个官方与媒体信源采集。摘要只转述原文内容,详情请看原文。 2026-11-11 全部新闻 · 按日期归档 · 全部 AI 新闻 · 今日

数据更新于:AI 新闻(北京时间)