注册    登录
专栏名称: 爱可可-爱生活
知名互联网资讯博主 北邮PRIS模式识别实验室陈老师
今天看啥
公众号rss, 微信rss, 微信公众号rss订阅, 稳定的RSS源
目录
今天看啥  ›  专栏  ›  爱可可-爱生活

【jax-flash-attn2：一个为JAX设计的灵活高效的F-20241025214630

爱可可-爱生活  · 微博  · AI  · 2024-10-25 21:46
    

文章预览

2024-10-25 21:46 本条微博链接 【jax-flash-attn2：一个为JAX设计的灵活高效的Flash Attention 2.0实现，支持GPU/TPU/CPU多后端和Triton/Pallas/JAX多平台】'jax-flash-attn2 - A flexible and efficient implementation of Flash Attention 2.0 for JAX, supporting multiple backends (GPU/TPU/CPU) and platforms (Triton/Pallas/JAX).' GitHub: github.com/erfanzar/jax-flash-attn2 #FlashAttention# #JAX# #GPUSupport# ………………………………

原文地址：访问原文地址
快照地址： 访问文章快照
总结与预览地址：访问总结与预览
推荐文章
爱可可-爱生活  ·  【[170星]cloudflare/ai-utils：为Clou-20250605213948
19 小时前
量子位  ·  抽象小视频秒变特效大片：原视频精髓不变角色环境任意换，Luma出品
昨天
爱可可-爱生活  ·  //@爱可可-爱生活：欢迎参与～-20250605080843
昨天
爱可可-爱生活  ·  【[36星]ncnn-android-ppocrv5：基于ncn-20250604182340
昨天
新智元  ·  昇腾+鲲鹏联手上大招！华为爆改MoE训练，吞吐再飙升20%，内存省70%
2 天前
常青藤爸爸  ·  优惠活动倒计时丨眼科大咖颠覆了我的认知！真正有助近视防控的营养竟是它！
1 年前
信达证券研究  ·  【信达金融】6月社融点评：金融总量指标或步入新常态
10 月前
湖北日报  ·  普京最新表态！
4 月前
中国非公立医疗机构协会  ·  关于举办商业健康保险助力医疗服务运营培训班的通知（第二轮）
1 月前
关于   移动版   ·   RSS之家   ·   卧龙AI搜索   ·   更多产品   ·  
今天看啥 - 公众号rss, 微信rss, 微信公众号rss订阅, 稳定的RSS源
© 2025 ~ 沪ICP备11025650号