├── .gitignore ├── LICENSE ├── README.md ├── figures ├── Figure_1_left_accuracy.png └── Figure_1_left_loss.png ├── grokking ├── cli.py ├── data.py ├── model.py └── training.py ├── requirements.txt └── sweep.yaml /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danielmamay/grokking/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danielmamay/grokking/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danielmamay/grokking/HEAD/README.md -------------------------------------------------------------------------------- /figures/Figure_1_left_accuracy.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danielmamay/grokking/HEAD/figures/Figure_1_left_accuracy.png -------------------------------------------------------------------------------- /figures/Figure_1_left_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danielmamay/grokking/HEAD/figures/Figure_1_left_loss.png -------------------------------------------------------------------------------- /grokking/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danielmamay/grokking/HEAD/grokking/cli.py -------------------------------------------------------------------------------- /grokking/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danielmamay/grokking/HEAD/grokking/data.py -------------------------------------------------------------------------------- /grokking/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danielmamay/grokking/HEAD/grokking/model.py -------------------------------------------------------------------------------- /grokking/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danielmamay/grokking/HEAD/grokking/training.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danielmamay/grokking/HEAD/requirements.txt -------------------------------------------------------------------------------- /sweep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danielmamay/grokking/HEAD/sweep.yaml --------------------------------------------------------------------------------