How to Tame Grokking: Representation Geometry as a Control Signal

This record contains the source code, experimental results, training logs, CSV files, figures, and supplementary materials used in the study: How to Tame Grokking: Representation Geometry as a Control Signal (arXiv:2607.11666). The archive includes implementations of Geometric Dimensionality Regularization (GeomDR), experiment configurations, raw and processed results, and scripts used to generate the figures and analyses reported in the paper. Related preprint: Kazanskii MA. How to Tame Grokking: Representation Geometry as a Control Signal. arXiv:2607.11666 (2026). DOI: 10.48550/arXiv.2607.11666.