TPU 架构与 Pallas Kernel 编程入门：从内存层次结构到 FlashAttention

本文通过四个复杂度递增的 kernel 展示了 Pallas 的核心编程模式

Deephub 11 天前 0 收藏

JAX性能优化实战：7个变换让TPU/GPU吃满算力

我们今天就来总结7个能够提高运行速度的JAX变换组合

Deephub 2026-01-04 20:34:54 0 收藏
登录可以使用的更多功能哦！ 登录
作者榜
...
资讯小助手

资讯同步

...
内容小助手

文章同步

...
Deephub

公众号：deephub-imba

...
奕凯

公众号：奕凯的技术栈