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