Grokking Replication Grokking original paper: https://arxiv.org/pdf/2201.02177.pdf Setup: pip install -r requirements.txt Running: python main.py Generating a graph python graph.py Accuracy curves from the grokking paper Accuracy curves generated with this repo