• Home
  • Technology
  • Gaming
  • Entertainment
  • World & Business
  • Science
  • Sports
  • AI
HomeTechnologyGamingEntertainmentWorld & BusinessScienceSportsAI
AI
Reaction

Student releases TPU kernels for modern linear attention variants

The student's instructor calls the kernels “impressively fast” and asks Google DeepMind about collaborating on model architecture research.

Han GuoHG
3 Sources, 3h ago, first seen 3h ago

TLDR

An instructor says one of their students released kernels for modern variants of linear attention on TPUs and calls them “impressively fast.” They say they still need TPUs to do model architecture research and ask Google DeepMind if it wants to collaborate.

Combined views

6

3 Sources, first seen 3h ago

22 reposts

Combined views

6

3 Sources, first seen 3h ago

22 reposts

Sentiment

Positive——Negative

Summary

Not enough discussion yet.

No sentiment analysis available yet.

Featured Source

Sentiment

Positive——Negative

Summary

Not enough discussion yet.

No sentiment analysis available yet.

3 Sources

Han Guo@HanGuo97RT @MayankMish98: Releasing FLAT: Flash Linear Attention on TPUs 🚀 Pallas training kernels for Linear Attention, Mamba-2 and GLA. On one T…3h
Joey Gonzalez@profjoeygOne of my students just released some impressively fast kernels for modern variants of linear attention on TPUs. Now we just need some TPUs to do the actual model architecture research, @GoogleDeepMind want to collaborate?1h
Ion Stoica@istoica05Pushing TPUs to their limits. FLAT hits 90% of peak HBM bandwidth (1.48 TB/s) and delivers large training speedups for Mamba-2 and GLA over standard H100 baselines. Great deep dive into kernel-level hardware optimization!1h
    • Home
    • Technology
    • Gaming
    • Entertainment
    • World & Business
    • Science
    • Sports
    • AI

    3 Sources

    Han Guo@HanGuo97RT @MayankMish98: Releasing FLAT: Flash Linear Attention on TPUs 🚀 Pallas training kernels for Linear Attention, Mamba-2 and GLA. On one T…3h
    Joey Gonzalez@profjoeygOne of my students just released some impressively fast kernels for modern variants of linear attention on TPUs. Now we just need some TPUs to do the actual model architecture research, @GoogleDeepMind want to collaborate?1h
    Ion Stoica@istoica05Pushing TPUs to their limits. FLAT hits 90% of peak HBM bandwidth (1.48 TB/s) and delivers large training speedups for Mamba-2 and GLA over standard H100 baselines. Great deep dive into kernel-level hardware optimization!1h
    Today's Rank

    —

    Not ranked yet

    Today's Rank

    —

    Not ranked yet