更多博客
博客进阶

你的模型能在 AWS Trainium 上运行吗?查一查就知道

Does Your Model Run on AWS Trainium? Look It Up

AI 导读

文章介绍一个 torch-neuronx 模型支持情况 Space,用于查询 Hub 上热门模型能否在 AWS Trainium 上运行。该 Space 的 Selected 视图展示按下载量和趋势分挑选的模型,每行给出结果、模型分片占用的 Trainium 核心数、运行实际耗时(含编译)和峰值设备内存;Trending 和 Downloads 视图按 Hub 自身排名逐任务列出结果,并对故意未测试的模型说明原因,还可选择 neuronx-cc 和 Transformers 的版本。这是继此前介绍 TorchNeuron 的文章之后的更新。

分级
值得看
收录时间
2026-10-26