跳到正文

i14 与 AWS 适配 Laplace 3B,以便在 Trainium2 上训练和服务

1篇报道 1个报道来源 3 小时前更新

先了解这件事

AI 综述

i14 与 AWS 将 i14 的 Laplace 3B 线性注意力模型适配到 AWS Trainium2,使用 TorchNeuron Native 通过原生 PyTorch 进行训练和服务,并报告训练速度约为每秒 120,000 个 token,推理吞吐量为每秒 11,544 个输出 token。 AWS 表示,其 GenAIIC 团队在一台 trn2.48xlarge 实例上使用全部 64 个逻辑 NeuronCore 训练了该模型,训练结果报告覆盖约 440 亿个 token。该模型为每个序列保持固定的 29.5 MB 循环状态。推理方面,团队在一台 trn2.48xlarge 实例上使用未经修改的 vLLM 引擎;报告的吞吐量是在 1,280 个并发序列下测得的。

AI 根据报道生成 · 2 小时前更新

报道时间线

沿着报道,了解事件的不同侧面。

10月21日
  1. 使用原生 PyTorch 在 AWS Trainium2 上训练和服务 i14 的 Laplace 3B

    i14 和 AWS 介绍了如何通过 TorchNeuron Native 使用原生 PyTorch,将 Laplace 3B 的线性注意力架构适配到 AWS Trainium2 上进行训练和服务。该模型为每个序列维持固定的 29.5 MB 循环状态,训练速度约为每秒 120,000 个 token,报告的结果覆盖约 440 亿个 token。在单个 trn2.48xlarge 实例上,同时处理 1,280 个序列时,推理速度达到每秒 11,544 个输出 token。

明早 08:00 把这样一份日报发到你邮箱

和上面这份一样:几件值得看的事。免费,点确认信之后才会开始发送,每封信都能一键退订。 看一份往期日报

本事件热度走势

还没有足够的连续观测数据,暂不绘制趋势。

继续阅读

最新日报 · 第 27 期TorchNeuron 为 AWS Trainium 上的 Hugging Face 模型带来原生 PyTorch 支持

正在热议

  1. 1Mistral 发布 Voxtral 3 语音模型、语音智能体 API 和基于 LiveKit 的 SDK
  2. 2DeepSeek 开源 DeepInfer 和 API 组件,但模型与硬件支持有限
  3. 3OpenAI 收购 Orchestra Bio,拓展药物研发 AI 工具
  4. 4谷歌开始向 Android Auto 推出 Gemini 个人智能功能,搭载 Google Built-In 的汽车也将在之后支持