跳到正文
原文
AGI Hunt· SonglinYang4·· 2 小时前AI 评分45

UC Berkeley 团队开源 FLAT,线性注意力 TPU 训练内核提速最高 3 倍

AI 导读

UC Berkeley、Stanford、Princeton 等机构研究者开源 FLAT(Flash Linear Attention on TPUs),这是首批面向 TPU 的线性注意力训练内核。

来源:AGI Hunt · agihunt.info