AWS 推理分析:一个 Strands Agent 背后的精度、内存与六大后端

邱宇AI 前沿📡 BestBlogs·AI高分精选⭐ 872026-10-13956 阅读💛 89 收藏
AWS 推理分析:一个 Strands Agent 背后的精度、内存与六大后端

📌 One-Sentence Summary

作者用同一个 Strands Agent 在六种 AWS 后端上运行 Gemma 4 E2B 等模型,用精度格式、内存带宽和推理引擎解释速度、成本与工具调用表现的差异。

📝 Summary

本文是“Inference on AWS”演讲的配套文章:同一个 Strands Agent 把同一问题发送到 Amazon Bedrock、SageMaker 端点、EC2 g6(NVIDIA L4)与 g5g(Graviton2 + T4G)上的 vLLM,以及手工移植 Gemma 4 的 Inferentia2 和 Trainium。同一个 Gemma 4 E2B 在不同后端上速度从 36 到 132 tok/s 不等,小时价格解释不了差距,关键在每个权重的位数、芯片读取速度和引擎对芯片的利用。基于 Google QAT checkpoint 的 int4 与 bf16 准确率相差不到 0.5 分,而从 bf16 直接舍入的 int8 损失 1.3 分;把 embedding 表压到 int4 可使 E2B 从 6.32 GiB 降到 2.85 GiB。g6 上的 vLLM 单请求 131.9 tok/s 最快、成本最低;SageMaker 每请求多约 0.24 秒、单 token 成本约为 g6 的两倍;Neuron 芯片带宽是 L4 的 2.0–2.7 倍,解码速度却只有其 0.28 倍,且尚无 vLLM 版本支持在 Neuron 上运行 Gemma 4。作者说明了测试范围:一次排练、每个后端三次计时请求。

💡 Main Points

同一模型速度差 3.7 倍

Gemma 4 E2B 在不同 AWS 后端上从 36 到 132 tok/s,取决于精度格式、内存带宽和推理引擎,而非小时价格。

QAT 量化几乎无损

基于 QAT checkpoint 的 int4 与 bf16 相差不到 0.5 分,bf16 直接舍入的 int8 损失 1.3 分;int4 embedding 把 E2B 从 6.32 GiB 压到 2.85 GiB。

g6 + vLLM 最快最省

单请求 131.9 tok/s;SageMaker 同 GPU 每请求多约 0.24 秒、单 token 成本约两倍。

带宽高不等于快

Neuron 芯片带宽是 L4 的 2.0–2.7 倍,解码速度仅 0.28 倍;目前没有 vLLM 版本在 Neuron 上支持 Gemma 4。

💬 Key Quotes

基于 Google QAT checkpoint 构建的 int4 模型与 bf16 的差距不到半分,而从 bf16 舍入的 int8 模型损失了 1.3 分。

g6 上的 vLLM 解码最快,单请求 131.9 tok/s。

Neuron 的带宽是 L4 的 2.0 到 2.7 倍,解码速度却只有其 0.28 倍。

📊 Article Meta

AI Screening: 87

Source: DEV Community: machinelearning

Author: xbill

Category: 人工智能

Language: 英文

Read Time: 9 min

Word Count: 2051

Tags:

AI 与智能应用 , AI 工程 , LLM 推理优化 , 云原生 / DevOps , 量化技术

#AI 与智能应用# AI 工程# LLM 推理优化# 云原生 / DevOps# 量化技术

文章评论(12)

星观澜22 小时前

这篇文章分析得很透彻,收藏了!

回复
暮临风1 天前

这个观点很中肯,深有同感。

回复
杨丽华23 小时前

看完了,收获满满,期待更多更新。

回复
山问津22 小时前

收藏了,以后慢慢研究。

回复
风倚栏21 小时前

点赞,必须点赞

回复
风倚栏20 小时前

赞同,实践出真知。

回复
林观澜20 小时前

很有价值的分享,感谢整理。

回复
墨沐雨20 小时前

实话,说的不明不白

回复
雪知秋20 小时前

这篇文章分析得很透彻,收藏了!

回复
陈皮话梅糖18 小时前

内容翔实,正好需要,先收藏再看。

回复
雪影17 小时前

很有价值的分享,感谢整理。

回复
空向阳16 小时前

实话,说的不明不白

回复