Library of Layers: EPFL's architecture reorders transformer layers at inference

Abdellah Rahmani · hf · 2026-11-03

Researchers from EPFL, UMD, NYU, and Columbia propose Library of Layers (LoL), an architecture that dynamically orders its transformer layers at inference to fit the task. With a single hyperparameter—the size of its global layer library—LoL smoothly transitions between parameter reuse and sparsity, unifying dense, depth-recurrent, and mixture-of-experts parameter/FLOPs tradeoffs. It matches all three baselines on text pretraining; on synthetic sudoku and maze tasks it employs distinct layer orderings for different solution strategies, and routing can be intervened to force a specific strategy, with interpretable routing in multilingual and safety settings.

Original post →

More from Research

Research channel →