跳到正文
热点事件历史事件

i14 在 AWS Trainium2 上实现 Laplace 3B 模型训练与部署

1 篇报道1 个报道来源20 天前更新

先了解这件事

AI 综述

i14 利用 TorchNeuron Native 在 AWS Trainium2 上实现了 Laplace 3B 线性注意力模型的训练与推理。该模型将推理状态固定为每序列 29.5 MB,在单台 trn2.48xlarge 实例上实现了约 120,000 tokens/s 的训练吞吐和 11,544 output tokens/s 的推理吞吐。

AI 根据报道生成 · 20 天前更新

报道时间线

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

10月21日
  1. AWS Machine Learning Blog精选
    i14 在 AWS Trainium2 上训练和部署 Laplace 3B 线性注意力模型

    i14 利用 TorchNeuron Native 在 AWS Trainium2 上实现了 Laplace 3B 线性注意力模型的训练与推理。该模型将推理状态固定为每序列 29.5 MB,在单台 trn2.48xlarge 实例上实现了约 120,000 tokens/s 的训练吞吐和 11,544 output tokens/s 的推理吞吐。