Back to Blog
Architecture and scaling overview

Loop Transformers: A Third Scaling Axis

Reframing reusable latent depth between parameter scaling and explicit reasoning tokens

July 22, 2026 · Technical Overview
This article synthesizes public papers and an author blog. Models, data, compute budgets, and evaluation protocols differ across studies, so the reported numbers illustrate within-paper findings rather than a unified leaderboard.

The two most familiar ways to scale a language model are both expensive.

The first is to add parameters and data during training, storing more knowledge and capability in the weights. The second is to generate more reasoning tokens at inference time, explicitly unrolling a longer search process. The former raises training, storage, and deployment costs. The latter increases latency, KV-cache usage, and sequential decoding work.

Loop Transformers introduce a third control knob between them: reuse the same Transformer parameters to update hidden states repeatedly, trading additional latent depth for capability without storing a fresh set of weights at every effective layer.

A standard Transformer applies

hl+1 = Fθl(hl)

whereas a looped model is closer to

h(r+1) = Fθ(h(r), x)

where the input x may be reinjected at every recurrence and r denotes the loop index. If a model stores L physical layers and executes each layer R times, it performs roughly L × R block executions while storing parameters for only about L layers.

The appealing claim is not that a small model becomes a large model for free. It is that three quantities can now be adjusted with greater independence:

  1. How many distinct transformations are stored: parameter count.
  2. How many times hidden states are refined: latent recurrent depth.
  3. How many intermediate results are written into the context: reasoning tokens.

My current view is that looping is a real scaling axis, but not free depth and not yet a universally validated replacement at frontier scale. Its value depends on three conditions holding at once: pretraining must teach a reusable iterative operator; recurrent dynamics must remain stable at useful depths; and the system must convert savings in parameter storage and communication into actual wall-clock gains.

1. What exactly does the third axis scale?

Scaling strategyPrimary resource addedWhere intermediate state livesTypical costPrimary benefit
Parameter scalingDistinct parameters and layersWeights and activationsModel memory, training FLOPs, deploymentKnowledge capacity and depth specialization
Explicit-token scalingGenerated sequence lengthContext and KV cacheDecoding latency, KV memory, per-token computeReadability, verification, tool use
Loop scalingExecutions of shared parametersHidden stateActivations, forward passes, latency, stabilityParameter reuse, continuous latent compute, adjustable depth

These approaches are not simple substitutes. Explicit chains of thought can be inspected, revised, and verified. Latent loops are compact but not inherently interpretable. Independent layers can specialize by depth; a shared layer must serve several stages of computation. Looping can reduce stored parameters, but it does not automatically reduce executed computation.

Any serious comparison should therefore report at least:

  • stored parameters;
  • active parameters, especially for MoE models;
  • effective block executions, roughly physical depth times loop count;
  • training and inference FLOPs;
  • activation memory, KV cache, latency, and throughput;
  • optimizer-step time or tokens per second, not only a theoretical FLOPs proxy.

A looped model may store fewer parameters while executing more block FLOPs. Conversely, it may have higher nominal FLOPs yet achieve better step time because it uses fewer pipeline stages, supports a larger microbatch, or reduces parameter communication.

2. A long intellectual lineage

Looping did not appear from nowhere.

The Universal Transformer already reused the same self-attention and transition modules across representation depth in 2018. With Adaptive Computation Time, different positions could consume different numbers of steps. It established the architectural viability of shared parameters and adaptive depth, along with a theoretical universality result under stated assumptions. It did not, however, answer the modern question: can a decoder-only LLM learn and exploit recurrent depth stably over hundreds of billions or trillions of pretraining tokens?

ALBERT later showed that cross-layer sharing could deliver substantial parameter efficiency in large-scale pretraining. Deep Equilibrium Models reframed an infinitely deep weight-tied network as a fixed-point problem, trained through root finding and implicit differentiation. Both loosened the assumption that every extra layer requires a new parameter set, but their main goals—parameter reduction and equilibrium computation—differ from finite, explicitly controllable recurrent depth in decoder language models.

The ICLR 2024 paper Looped Transformers are Better at Learning Learning Algorithms moved the idea closer to reasoning. On synthetic in-context learning tasks, a one-layer looped GPT-2 could approach a 12-layer standard Transformer. The two models used about 0.79M and 9.48M parameters respectively, so the looped model had roughly one twelfth as many parameters. The method reinjected the original prompt at every recurrence and applied losses over several of the final loop steps rather than supervising only the last one.

The important lesson was not that one layer had categorically beaten twelve. It was that weight sharing may be a more natural inductive bias when a task calls for repeated applications of a similar update rule. The model behaves more like an iterative algorithm. The same work also found boundaries: an error floor remained, excessive recurrence slowed or destabilized training, and the learned procedure did not generalize as a fully distribution-independent algorithm.

In 2025, Huginn and Ouro brought recurrent depth into the main path of language-model pretraining. In 2026, Fully Looped Transformer, STARS, The Readout Blind Spot, and Loopie added important pieces on stability, failure mechanisms, and system efficiency.

3. Two main architectures: model-loop and layer-loop

Suppose a physical model stores three layers and executes each one twice:

model-loop: L1 → L2 → L3 → L1 → L2 → L3
layer-loop: L1 → L1 → L2 → L2 → L3 → L3

In a model-loop, the entire physical stack acts as a recurrent core. The output of one full pass returns to the first layer. Huginn's prelude–core–coda design and Ouro both belong broadly to this family. The loop boundary is explicit, making it convenient to control total recurrence, learn exit gates, or reinject the input at each pass.

In a layer-loop, each physical layer runs multiple times before execution moves to the next layer. Loopie adopts this pattern and offers two reasons to take it seriously.

The first is systems locality. A model-loop sends computation from the last pipeline stage back to the first, creating cyclic dependencies and pipeline bubbles. Layer-loop repetition remains within the current stage and fits existing parallel-training stacks more naturally.

The second is a more local parameter-sharing pattern. In a 48-layer model with two full-stack loops, the third physical layer is used at effective depths 3 and 51. Those invocations receive states that have undergone very different amounts of preceding computation, potentially asking one shared transform to serve both early- and late-depth roles. A layer-loop reuses a layer at adjacent effective depths, where its role may be more coherent.

This argument needs calibration. Prior work shows that lower, middle, and final Transformer layers do not represent information uniformly, making heterogeneous demands across distant effective depths plausible. It does not directly prove gradient conflict. “Natural parameter sharing” is best treated as an empirically motivated design hypothesis, not a settled mechanism.

Loopie's long training curves add supporting evidence: layer-loop starts slightly behind model-loop and overtakes after roughly 1.2T tokens. But this result still depends on architecture, optimization, and system conditions. One curve does not establish universal layer-loop superiority.

4. Four stages that should not be conflated

Pretraining, post-training, test-time compute, and test-time training are often mixed together in loop discussions. They are four different experiments.

StageAre weights updated?Does loop count vary?Main objectiveExamples
Loop-aware pretrainingYesFixed or sampled during trainingLearn a stable reusable recurrent operatorHuginn, Ouro, Loopie
Post-training (SFT/RL)YesUsually keeps the pretrained structureShape reasoning, instruction following, and behaviorOuro Thinking, Loopie SPT/RL
Frozen-weight depth scalingNoYesTrade more forward recurrences for capabilityHuginn, FLT, STARS
TTT / test-time adaptationTemporarily, then resetMay be fixed or changedAdapt a small parameter subset to the current prompt or taskLoop-TTT blog experiment

The first stage is foundational. If a model never learned recurrent dynamics during pretraining, mechanically rerunning an ordinary checkpoint's layers at inference will not usually produce stable gains. Huginn samples recurrence counts during training. Ouro trains predictions at multiple steps and learns adaptive exits. Loopie is pretrained from scratch with two layer-local executions. None is a training-free retrofit of a standard checkpoint.

The second stage shapes behavior after the base model is trained. Loopie calls its large target-only supervised stage Supervised Pre-Training (SPT), but in lifecycle terms it occurs after base pretraining, followed by math and code RL. The final Thinking model's competition results combine looped pretraining, SPT, RL, and generate–verify–refine. They cannot all be attributed to layer-looping.

The third stage freezes the weights and changes only inference depth. Some papers loosely call this test-time adaptation, which can be misleading. The relevant Fully Looped Transformer experiments vary loop depth with frozen weights; no gradient-based adaptation occurs on the test sample.

The fourth stage is test-time training in the strict sense: compute a loss from the current input, perform one or a few gradient updates, and reset afterward. It incurs backward-pass and optimizer costs and should not be compared with extra forward loops under an unmatched latency budget.

5. Huginn: learning recurrent depth during pretraining

Huginn is a useful starting point for modern looped LLMs. It divides the network into a prelude, a recurrent core, and a coda. The prelude introduces token representations into the loop, the core updates latent state repeatedly, and the coda produces logits. The reported 3.5B model uses a 2–4–2 physical-layer layout. Repeating the four-layer core an average of 32 times yields an effective depth of about 2 + 4 × 32 + 2 = 132 block executions.

The core contribution is not copying four layers 32 times. Training samples the recurrence count from a distribution and backpropagates through only the final subset of iterations. The model therefore cannot rely entirely on one absolute value of r; it must learn an update process that continues to function across a range of depths.

Huginn also shows how much more fragile recurrence becomes at scale. Early runs experienced representation or recurrence collapse. Sandwich RMSNorm, repeated input injection, and initialization choices became essential. The final model trained on about 800B tokens. Around the trained depth range, additional recurrence improved several tasks, and the paper explored per-token adaptive compute based on state convergence, KV-cache sharing, continuous-CoT warm starts, and self-speculative decoding.

These results support two claims. First, latent recurrence can be learned in realistic language-model pretraining, not only in synthetic regression. Second, hidden trajectories display convergence, orbit-like motion, and gradual state changes. But calling those trajectories an “invisible chain of thought” remains an interpretation, not an identified causal mechanism. We can verify repeated state transformation and task gains; we cannot read a human-level reasoning step out of every recurrence.

6. Ouro: parameter efficiency is real, but shared depth is not independent depth

Ouro develops model-looping into a more complete training pipeline. The same physical stack is applied repeatedly, with an exit gate at each recurrence. The objective includes entropy regularization to prevent the gate from collapsing to the deepest step. A later phase freezes the language model and sharpens depth allocation based on marginal loss improvement.

Ouro provides a strong signal for the approach. Its 1.4B and 2.6B models approach larger models on some benchmarks, summarized by the paper as roughly 2–3× parameter efficiency. Controlled synthetic-knowledge experiments separate storing facts from manipulating them: extra loops do not create additional factual storage, but they improve multi-hop composition and knowledge manipulation. That is consistent with the third-axis view—recurrence adds operations over existing representations rather than new storage slots.

One appendix result is more important than the headline: when model size, training tokens, and recurrent executions are controlled, a standard Transformer with independent layer parameters still outperforms the LoopLM. The gap initially grows with deeper recurrence and narrows somewhat at larger model scale.

This reframes the real question. Looping does not claim that shared depth is intrinsically more expressive than untied depth. It asks:

Under constraints on parameter storage, communication, deployment, or system efficiency, can shared depth offer a better overall exchange rate?

Independent layers are free to specialize by depth and are naturally more expressive. Looping must earn its advantage through parameter reuse, increased training scale, adaptive depth, or better systems behavior—not by assuming that sharing has no cost.

7. Why deeper loops fail: a dynamical-systems problem

Writing the recurrent core as

ht+1 = Φθ(ht; x)

makes the issue explicit. Every additional loop applies the same nonlinear dynamical system again. The trained depth range covers only a segment of its trajectory. Beyond that segment, the state may converge, oscillate, drift, explode, or fall into an undesirable attractor.

7.1 Residual explosion and oscillating gradients

Fully Looped Transformer identifies two characteristic failure modes: early gradient oscillation and residual explosion as loop count rises. Its original looped baselines collapse at 9 or 12 loops in some small-model settings.

The paper proposes two parameter-free changes. Fully Looped Architecture exposes the previous loop's output to every physical layer rather than injecting it only at the loop boundary. Attention Injection uses the previous loop state as the query and the current layer state as key/value while reusing existing projections. In 127M/318M experiments, these changes remain stable through 12 loops. In one reported average, the original looped model falls from 40.52 at three loops to 36.56 at six, while the Fully Looped model rises from 39.90 at three loops to 41.72 at nine.

The paper's 13.2% figure is a relative improvement over its original looped baseline in a particular six-loop setting, not a universal 13.2-point gain. The models are only hundreds of millions of parameters. The work is valuable as a diagnosis and architectural remedy, not as a completed frontier-scale extrapolation.

7.2 Bounded state is not necessarily useful state

STARS frames local stability through the Jacobian spectral radius. Near a state, if ρ(J) < 1 for J = ∂Φ/∂h, perturbations contract locally; sustained values above one can amplify errors through recurrence.

The method estimates the spectral radius with one power iteration and combines Jacobian Spectral Radius Regularization with random loop sampling. On Ouro-1.4B fine-tuned on 400K NuminaMath examples, the base model's average drops from 67.41 at r=4 to 55.09 at r=8; ordinary SFT falls from 70.46 to 52.97; STARS falls more gently from 74.18 to 65.55. It substantially reduces deep-loop degradation, but it does not make accuracy monotonic in depth.

The lesson is that stability is necessary but not sufficient. A dynamical system can converge reliably to an unhelpful fixed point, or remain bounded while collapsing distinct inputs into poorly separated attractors.

7.3 The readout blind spot in dense supervision

A subtler failure comes from normalization. Many models attach a cross-entropy loss to every loop, which appears to provide dense supervision. But with RMSNorm or LayerNorm before the logits, the readout is approximately invariant to the hidden state's overall scale. Scaling h changes little after normalization, so the immediate CE loss supplies almost no radial gradient.

The pre-norm recurrent path still receives the unnormalized state in the next loop. Scale can therefore accumulate while every per-loop readout remains blind to it. The Readout Blind Spot reports final hidden norms of 39,207 and 56,051 in 44M/129M models using per-loop CE with normalized readouts. Raw, scale-visible readouts or explicit norm penalties keep norms in the tens and recover perplexity improvements as loop count increases.

This explains why “loss at every loop” may still fail to train a useful recurrent process. It also reveals an evaluation trap: if an exit gate always halts early, the model may have learned efficient adaptive compute—or later loops may simply have no value. Ouro's inter-loop RMSNorm mitigates this specific problem, but the main causal evidence remains small-scale; its 1.4B experiment is closer to a sanity check.

8. Loopie: layer-looping as model–system co-design

Loopie is one of the clearest cases for asking whether recurrence can produce real training gains. It does not merely run every layer of a 48-layer model twice. It roughly halves stored depth, restores effective execution depth with R=2 layer-loops, and reinvests memory savings into width, depth, and batch organization.

The paper's Qwen3-like reference is approximately 30B-A3B with 48 layers and hidden size 2048. The selected Loopie is about 20B-A2B with 27 layers and hidden size 2304, each layer executed twice. A block-compute proxy places Loopie at about 1.42× the reference's nominal compute, while an activation proxy at the reference microbatch is about 0.63×. Fewer physical layers allow the microbatch to double and gradient-accumulation steps to halve, bringing measured optimizer-step time close to the baseline.

“Compute matched” therefore means matched hardware, tokens, updates, and approximately step time—not identical theoretical FLOPs. That is an important engineering result and a reminder that neither parameter count nor FLOPs alone captures efficiency.

In pretraining, Loopie overtakes the vanilla baseline after roughly 600B tokens. The model-loop/layer-loop comparison reverses in favor of layer-loop after about 1.2T tokens. The paper also builds a scaling ladder from 0.15B to 1B active parameters with approximately wall-clock-matched pairs and observes a Loopie advantage at every rung. A controlled 6B-A0.6B ablation becomes worse when the layer-loop schedule is removed, supporting the view that the result is not merely hidden extra compute.

Three limits remain. First, the public ladder's largest baseline has only about 1B active parameters, still far from frontier scale. Second, the final 20B-A2B Thinking model adds roughly 2T supervised tokens, math/code RL, and generate–verify–refine at competition time. Third, R=2 is selected because marginal recurrence returns decline under a fixed pretraining budget; it is not a universal optimum.

Loopie's strongest conclusion is therefore not that looped models now beat large models everywhere. It is that layer-looping can turn parameter sharing into activation, pipeline, and batch efficiency, then use model–system co-design to move the wall-clock frontier.

9. Evaluate with a scaling ladder, not one large run

Loop Transformers particularly need scaling ladders. Gains from recurrence at small scale may change with width, data, training duration, and hardware parallelism. A single large run cannot cleanly separate architectural gains from extra FLOPs or a stronger training recipe.

An informative ladder should:

  1. keep data distribution, tokenizer, optimizer, training stages, and evaluation consistent across several scale rungs;
  2. state whether it matches theoretical FLOPs, tokens, step time, total wall-clock, or hardware budget;
  3. report stored and active parameters, physical layers, loop count, and effective block executions separately;
  4. include a non-looped control and ablations for “extra compute only” and alternative sharing patterns;
  5. estimate trends at smaller scales, then calibrate extrapolation with a limited number of larger checkpoints.

TPP usually means tokens per parameter, D/N. For dense models, N is relatively unambiguous. For MoE, a paper must state whether the denominator is total or active parameters. Loopie's ladder uses roughly 1,000 active-parameter tokens per active parameter, except for a token cap at the largest rung. Fixed TPP keeps models at roughly comparable data-to-parameter training stages, but it is not fixed total FLOPs and does not imply fixed wall-clock.

If a team uses a composite CE (compute-efficiency) score, it should publish the formula, baseline, denominator, and aggregation. CE is not a cross-paper standard unit. The most robust public presentation remains a performance–compute frontier under an explicit resource constraint.

For looped models, I would report three frontiers:

  • quality versus stored parameters, for parameter efficiency;
  • quality versus training wall-clock or FLOPs, for training efficiency;
  • quality versus inference latency or executed FLOPs, for inference efficiency.

Improving only the first may amount to trading compute for memory. Improving all three would be much closer to an architectural breakthrough.

10. Increasing loops at inference: depth is not unlimited

The most attractive story is to train once and allocate more loops to harder problems at inference. Public evidence supports adjustment within or near the trained range; it does not support arbitrary extrapolation.

Huginn samples recurrence depths during training. Ouro learns exit gates. Fully Looped Transformer and STARS extend the stable range through architecture and regularization. Yet several experiments show the same pattern: adding loops helps up to around the trained depth, then performance fluctuates, degrades, or collapses.

This differs from explicit chain-of-thought scaling. Every generated token writes a new state into the context and receives direct next-token supervision. A latent loop repeatedly transforms the same state through a shared dynamical system. One more recurrence is not a free extra thought; it moves the state farther from the trajectory covered during training.

A more rigorous deployment policy would:

  • treat loop count as a training condition, not merely an inference hyperparameter;
  • calibrate depth–quality–latency curves by task;
  • base exit decisions on verifiable marginal value, not only apparent state convergence;
  • label depths beyond the trained maximum as extrapolation;
  • compare loops with explicit reasoning tokens, verifiers, and self-consistency under matched latency.

11. TTT is a separate path: update parameters instead of only adding depth

Alvin Zhou's Loop-TTT blog offers an illuminating, easily misread experiment. Using the Ouro-1.4B base model at its trained depth r=4, the author performs one prompt-entropy gradient update on 97 RMSNorm scale vectors—198,656 scalars, about 0.014% of the model—and resets after each batch.

On 500 GSM8K four-shot examples, accuracy rises from 0.766 to 0.848, an 8.2-point gain. Simply increasing the loop count from r=4 to r=8 yields only 0.694. A MATH-500 subset gains about three points, while an MMLU subset is nearly flat. The result suggests that a tiny temporary calibration of a shared recurrent operator may be more effective than blindly executing that operator more times.

The most informative result is the author's own counter-analysis. Almost all of the gain comes from the four demonstrations shared across questions rather than question-specific information. Updating on exemplars alone recovers about +7.0 of the full gain; a fixed cached update vector recovers most of it; update directions across batches have cosine similarity 0.99. The more accurate description is therefore cacheable prompt or task calibration, not per-question online learning.

The study is exploratory. Hyperparameters were tuned while observing GSM8K test accuracy; there is no comparison with a stronger post-trained Ouro; TTT requires forward, backward, and optimizer work without a strict latency match; and the same update is nearly neutral on Qwen2.5-3B. It is a promising side branch, not evidence that loop pretraining is already mature.

12. What the evidence supports today

ClaimCurrent evidenceWhat remains missing
Shared parameters can learn iterative update rulesSynthetic ICL, Huginn/Ouro pretrainingMechanistic validation across broader tasks and larger scales
Loops can improve parameter efficiencyOuro and parts of Loopie's comparisonsFrontier-scale replication with matched data and recipe
Shared depth can replace independent depthOnly partially; Ouro's controlled standard model remains strongerWhere the exchange rate becomes favorable
More test-time loops can improve qualityPositive results within trained depth rangesReliable extrapolation and per-sample routing
Stability failures can be mitigatedFLT, STARS, readout fixesLarge-scale, long-run, cross-domain validation
Layer-loop is generally better than model-loopLoopie systems analysis and long training curvesBetter-controlled replication and direct gradient evidence
Looping reduces real training costLoopie model–system co-designPortability across clusters, parallel strategies, and scales
TTT is a general inference enhancementSignal in one exploratory casePreregistered, cost-matched, cross-model evaluation without shared demos

Three findings appear most robust.

First, parameter sharing is not computation sharing. Looping reduces the number of distinct weights, not the operations executed per token. Its earliest gains are often in parameter memory, communication, and activation organization. Whether total time falls depends on the system.

Second, recurrent depth must be modeled during training. Random depth, input injection, per-loop supervision, state normalization, exit gating, and dynamics regularization are not decorative details. They determine whether the model learns an operator that can keep running.

Third, a shared operator creates functional-conflict and dynamical risks. Distant effective depths may demand different transforms; normalized readouts may ignore state scale; and extra loops outside the trained range may overthink. These are central architectural problems, not incidental tuning noise.

13. My view: conditional optimism

The most promising Loop Transformer is unlikely to compress a 100-layer model into one layer repeated 100 times. A more practical design will retain enough untied structure while selectively sharing the computations that can form a stable iterative process. Prelude/coda layers, grouped recurrence, layer-loops, partially untied layers, recurrent MoE, and hybrids of explicit-token reasoning with latent loops may all be more viable than indiscriminate full-model recurrence.

The next research questions are concrete:

  1. What should be shared? Which layers support local reuse, and which require depth specialization? Can natural sharing be tested directly with representation similarity, gradient angles, or functional probes?
  2. How long should the model loop? Can marginal value of compute be calibrated rather than delegated to an exit gate that may learn shortcuts?
  3. How do we preserve useful dynamics? Which measurements of spectral radius, state scale, direction, and attractor structure predict generation quality?
  4. How should recurrence interact with MoE? Should loops share experts, route to different experts, or allow routing itself to evolve across recurrence?
  5. How should systems gains be accounted for? Training and inference results should include step time, throughput, activation, KV, communication, and energy—not only stored parameters.
  6. How should latent and explicit reasoning cooperate? Which computation belongs in compact continuous state, and which steps should be written as tokens for verifiers, tools, or users?

If these questions are answered on reliable scaling ladders, model scaling will move beyond a one-dimensional discussion of parameter count. We will instead reason over a resource surface: how many weights to store, how many latent updates to execute, and how much explicit reasoning to generate.

That may be the real significance of Loop Transformers. They do not end scaling; they turn it from a line into a design space with at least three dimensions.


References

  1. Dehghani et al., Universal Transformers, 2018 / ICLR 2019.
  2. Lan et al., ALBERT: A Lite BERT for Self-supervised Learning of Language Representations, 2019 / ICLR 2020.
  3. Bai et al., Deep Equilibrium Models, 2019 / NeurIPS 2019.
  4. Yang et al., Looped Transformers are Better at Learning Learning Algorithms, ICLR 2024.
  5. Geiping et al., Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approach, 2025.
  6. Zhu et al., Scaling Latent Reasoning via Looped Language Models (Ouro), 2025.
  7. Fu et al., Simply Stabilizing the Loop via Fully Looped Transformer, 2026.
  8. Yang et al., Stabilizing Recurrent Dynamics for Test-Time Scalable Latent Reasoning in Looped Language Models (STARS), 2026.
  9. Sharma and Vu, Dense Supervision Is Not Enough: The Readout Blind Spot in Looped Language Models, 2026.
  10. Gao et al., Loop the Loopies! (Loopie), 2026.
  11. Alvin Zhou, Loop deeper, or adapt? Test-time training in looped transformers, 2026.