NVIDIA Technical Blog 发布于 10/01 04:54

Deploying an HSTU Generative Recommender with NVIDIA Dynamo-Triton

(翻译)使用 NVIDIA Dynamo-Triton 部署 HSTU 生成式推荐系统

查看原文

资讯摘要

Generative recommender (GR) systems are emerging as a powerful new approach for large-scale personalization. Instead of treating recommendation as a set of…

AI 摘要

英伟达技术博客介绍如何把 HSTU 生成式推荐模型从 PyTorch 开发迁移到生产推理:用 PyTorch AOTI 提前编译并导出模型,结合 FlexKV 支持的 KV 缓存与 NV Embedding Cache 降低显存占用和重复计算,再由 Dynamo-Triton 提供线上服务。在 RTX PRO 6000 Blackwell 上、100% GPU KV 缓存命中时,三层与八层 HSTU 模型延迟最高分别降低 4.47 倍和 5.93 倍。

AI 问答

HSTU 生成式推荐模型在线上服务时面临什么挑战?

每请求都要处理很长的用户历史序列,需要在不重算全部键值状态的前提下实现低延迟推理。

Dynamo-Triton 的完整部署流程包含哪些步骤?

构建自定义算子与运行库、用 PyTorch AOTI 导出 HSTU 排序模型、启动 FlexKV 支撑的 KV 缓存服务、用原生 C++ 校验导出产物、最后用 Dynamo-Triton 提供推理服务。

KV 缓存带来多大的性能提升?

在 RTX PRO 6000 Blackwell 工作站版 GPU、动态 batch size 8、GPU KV 缓存命中率 100% 的条件下,三层 HSTU 模型最高提速 4.47 倍,八层模型最高提速 5.93 倍。

打开原文

更多资讯

巧合?Ledger 用户被盗9000万 经销商两个月前刚易主金色财经 · 10/10 11:27用AI降低激光焊使用门槛,这家公司完成数千万元融资|36氪首发36氪 · 10/10 10:26旅行成为个人 Agent 的高频场景,「豆包工作」把几十小时攻略浓缩成 PPT人人都是产品经理 · 10/10 17:29美债抛售红灯:期限溢价破十年高位 脱离美联储路径金色财经 · 10/10 08:17奈飞据悉正进行组织重组,预计将影响约5%的员工36氪 · 10/09 19:43大模型通用答案没有考虑每个人需求金色财经 · 10/10 09:0224 天估值暴涨 37 倍,这就是 AI 圈年度爽文爱范儿 / APPSO · 10/10 17:11为什么 AI 产品需要体验评测人人都是产品经理 · 10/10 15:31Telegram钱包全面开放 CFTC将事件合约纳入互换定义金色财经 · 10/10 07:54WorkBuddy国际版实测,双持打开新世界人人都是产品经理 · 10/10 15:27Let's Encrypt 从 2027 年起切换到有效期为 64 天的证书Solidot · 10/09 23:12当年「字节投毒实习生」田柯宇,估值 2 亿美元,要挑战李飞飞做世界模型雷峰网 · 10/09 18:34独家丨阿维塔酝酿高层调整:董事长王辉已开始负责长安海外业务雷峰网 · 10/10 15:47机构:预估2031年VCSEL市场需求量将有望达到6.5亿颗36氪 · 10/10 12:52:fire::fire:开源接口平台v6.1发布,支持数据插件开源中国 · 10/10 09:51商务部欧洲司负责人解读中欧贸易投资磋商机制第二次例会成果36氪 · 10/10 13:35Cloudflare 让决策模型打起了价格战:Clef 降价、提速、上多模态开源中国 · 10/10 16:25今年白云机场口岸出入境客货航班超10万架次,创历史同期新高36氪 · 10/10 13:46商务部:中欧双方将继续探讨在世贸组织规则框架下,降低部分商品关税的可能性36氪 · 10/09 19:20Insights4vc:Rain 的 19.5 亿美元估值靠什么撑起金色财经 · 10/09 17:22How Does Life Unfold? A Landscape Metaphor Comes Into Its Own.Quanta Magazine · 10/09 22:03Citrini:区块链终于迎来它的时刻 怎么选择投资组合金色财经 · 10/09 18:34深交所:9月28日至10月9日,共对141起证券异常交易行为采取了自律监管措施36氪 · 10/09 19:57华尔街新高藏隐忧金色财经 · 10/09 22:27多部门:汽车生产企业不得作虚假、夸大或引人误解的宣传36氪 · 10/10 17:25人人皆可为师:全球首个物理AI数据众包服务平台“觅蜂派”正式发布雷峰网 · 10/10 11:45澳门大学预计未来5年学生规模逾2.2万人36氪 · 10/10 15:52美债失控狂飙:冲击到底有多大?金色财经 · 10/09 21:25什么样的代币化美股才是你需要的?金色财经 · 10/10 22:27人社部:将构建技能导向的薪酬分配制度,完善治理欠薪长效机制36氪 · 10/10 11:51