Omnigrok: Grokking Beyond Algorithmic Data - View it on GitHub