新PyTorch API:几行代码实现不同注意力变体,兼具FlashAttention性能和PyTorch灵活性
新PyTorch API:几行代码实现不同注意力变体,兼具FlashAttention性能和PyTorch灵活性用 FlexAttention 尝试一种新的注意力模式。
来自主题: AI资讯
10383 点击 2024-08-10 18:05
搜索
用 FlexAttention 尝试一种新的注意力模式。
不开颅,把 ChatGPT 装进脑子里?
七年前,论文《Attention is all you need》提出了 transformer 架构,颠覆了整个深度学习领域。
Tenstorrent推AI芯片挑战昂贵HBM,追求成本效益。
最近,Latent Space发布的播客节目中请来了Meta的AI科学家Thomas Scialom。他在节目中揭秘了Llama 3.1的一些研发思路,并透露了后续Llama 4的更新方向。
用来运行 Llama 3 405B 优势明显。
740 TFLOPS!迄今最强 FlashAttention 来了。
时隔一年,FlashAttention又推出了第三代更新,专门针对H100 GPU的新特性进行优化,在之前的基础上又实现了1.5~2倍的速度提升。
大模型训练推理神作,又更新了!
尤洋团队新作,首个基于DiT的实时视频生成方法来了!