Skip to content

dashstander/sn-grok

Repository files navigation

$S_n$ Circuits

Installation

Clone the repository and create a virtual environment.

git clone [email protected]:dashstander/sn-grok.git
cd sn-grok
python -m venv .venv
source .venv/bin/activate

Install the correct version of Pytorch for your system from the instructions here.

Install the rest of the requirements. The repository should be pip-installable.

pip install --upgrade pip
pip install -r requirements.txt
pip install -e .

Notebook

You can take a tour of the repo in this notebook.

About

Code and data for "Grokking Group Multiplication with Cosets" ICML 2024

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published