— saved image
Progyan @plugyawn · 10h If looping works (and it seems like we're only a little bit away), and softmax attention doesn't blow up memory in the process, I think it is trivial to imagine recirculation on an ASIC being a pathway to getting 5-10X the effective depth with barely any wallclock overhead. [embedded figure from a paper:] Figure 4: (a) Unrolled loop transformer and (b) unrolled recirculation transformer. The open colored rectangles depict state propagation. In the looped transformer, strict state propagation moves upward in the stack, whereas in recirculation, state propagation can continue indefinitely in the same layer of the stack.
Note from Claude Sonnet 5
A tweet about transformer architecture scaling, arguing that if 'looping' works without softmax attention blowing up memory, recirculation on an ASIC could give 5-10x effective depth with little wallclock overhead. Embedded is Figure 4 from an ML paper: two side-by-side diagrams (a) and (b) of grids of small rectangles representing an 'unrolled loop transformer' versus an 'unrolled recirculation transformer,' with colored state markers (state 1-4 in green/blue/purple/red) showing how state propagates upward through stacked layers in the looped case versus staying within the same layer indefinitely in the recirculation case.
transformersai architectureasicmodel scalingloopingtwitter discourse