Skip to content

Repository files navigation

$S_n$ Circuits

Installation

Clone the repository and create a virtual environment.

git clone git@github.com: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

Stars

4 stars

Watchers

1 watching

Forks

Releases

Packages

Used by

Contributors

Languages