Oossa

i14在AWS Trainium2上训练Laplace 3B,速度达每秒12万tokens

这款30亿参数的线性注意力模型在trn2.48xlarge实例上的训练速度达每秒120,000 tokens,服务吞吐量为每秒11,544 tokens。

简讯作者: Oossa 发布日期: 1 分钟阅读

i14使用AWS Trainium2和TorchNeuron Native训练并部署其线性注意力模型Laplace 3B。在单台trn2.48xlarge实例上,该模型的训练速度约为每秒120,000 tokens,并能同时处理1,280个序列,每秒生成11,544个输出tokens。无论上下文长度如何,模型架构都能将每个序列的推理状态固定在29.5 MB。

为什么重要

内存占用可预测,意味着长期运行的AI代理能够处理大得多的文本,而成本不会随之增加。

这篇文章有帮助吗?
分享

继续阅读

Oossa · 新闻简报

一周 AI,讲明白

每周一,用通俗的语言讲清值得知道的新闻。免费,无垃圾邮件。