A replication and visualization suite for "Progress Measures for Grokking via Mechanistic Interpretability" (Nanda, Chan, Lieberum, Smith & Steinhardt, ICLR 2023). Trains a minimal transformer on ...