Decoding the black box of LLMs: A comparative analysis of Logit Lens vs. Tuned Lens to interpret intermediate Transformer layers in GPT-2.
-
Updated
Apr 2, 2026 - Jupyter Notebook
Decoding the black box of LLMs: A comparative analysis of Logit Lens vs. Tuned Lens to interpret intermediate Transformer layers in GPT-2.
From-scratch PyTorch implementation of the Tuned Lens (Belrose et al., 2023) — learned per-layer affine probes that sharpen intermediate transformer predictions beyond the raw logit lens.
Code and artifacts for The Convergence Gap: when instruction-tuned models settle on next-token predictions.
Cross-architecture mechanistic interpretability toolkit — first OSS Mamba SSM state extraction. Works on transformer + SSM + hybrid models with unified API.
Research workspace for model diffing between pretrained and post-trained language models.
To associate your repository with the tuned-lens topic, visit your repo's landing page and select "manage topics."