├── .gitignore ├── README.md ├── data.py ├── main.py ├── media ├── grokking.png ├── grokking_black_spines.png └── grokking_white_spines.png ├── models.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stockeh/mlx-grokking/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stockeh/mlx-grokking/HEAD/README.md -------------------------------------------------------------------------------- /data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stockeh/mlx-grokking/HEAD/data.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stockeh/mlx-grokking/HEAD/main.py -------------------------------------------------------------------------------- /media/grokking.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stockeh/mlx-grokking/HEAD/media/grokking.png -------------------------------------------------------------------------------- /media/grokking_black_spines.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stockeh/mlx-grokking/HEAD/media/grokking_black_spines.png -------------------------------------------------------------------------------- /media/grokking_white_spines.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stockeh/mlx-grokking/HEAD/media/grokking_white_spines.png -------------------------------------------------------------------------------- /models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stockeh/mlx-grokking/HEAD/models.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stockeh/mlx-grokking/HEAD/requirements.txt --------------------------------------------------------------------------------