Fast TPU kernels for linear attention released; team pitches DeepMind on collaboration

gabriberton · x · 2026-10-14

A student of Prof. Joey released impressively fast TPU kernels for modern variants of linear attention. The team says it now just needs TPU access to do actual architecture research and has publicly asked Google DeepMind to collaborate. Gabrielle Berton reposted hoping relevant colleagues see it.

Original post →

More from Infra

Infra channel →