Alternatives and similar repositories for grokking:

Users that are interested in grokking are comparing it to the libraries listed below

• danielmamay / grokking
  Implementation of OpenAI's 'Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets' paper.
  ☆36Updated last year
• KindXiaoming / Omnigrok
  Omnigrok: Grokking Beyond Algorithmic Data
  ☆55Updated 2 years ago
• teddykoker / grokking
  PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆36Updated 3 years ago
• edwardjhu / TP4
  Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)
  ☆62Updated 3 years ago
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆48Updated last year
• TomFrederik / unseal
  Mechanistic Interpretability for Transformer Models
  ☆50Updated 2 years ago
• Ping-C / optimizer
  This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…
  ☆36Updated 2 years ago
• AllanYangZhou / universal_neural_functional
  ☆49Updated last year
• mechanistic-interpretability-grokking / progress-measures-paper
  ☆62Updated 2 years ago
• bilal-chughtai / rep-theory-mech-interp
  ☆26Updated last year
• aks2203 / easy-to-hard
  Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"
  ☆59Updated 3 years ago
• p-lambda / incontext-learning
  Experiments and code to generate the GINC small-scale in-context learning dataset from "An Explanation for In-context Learning as Implici…
  ☆105Updated last year
• ejmichaud / grokking-squared
  ☆25Updated 2 years ago
• berlino / seq_icl
  ☆51Updated 11 months ago
• GFNOrg / gfn-lm-tuning
  ☆175Updated last year
• redwoodresearch / Easy-Transformer
  ☆114Updated 8 months ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆136Updated last year
• timaeus-research / devinterp
  Tools for studying developmental interpretability in neural networks.
  ☆88Updated 3 months ago
• understanding-search / maze-transformer
  This repo is built to facilitate the training and analysis of autoregressive transformers on maze-solving tasks.
  ☆27Updated 7 months ago
• locuslab / edge-of-stability
  ☆67Updated 4 months ago
• wattenberg / superposition
  Code associated to papers on superposition (in ML interpretability)
  ☆27Updated 2 years ago
• ethancaballero / broken_neural_scaling_laws
  Code Release for "Broken Neural Scaling Laws" (BNSL) paper
  ☆58Updated last year
• lee-ny / teaching_arithmetic
  ☆79Updated last year
• jbloomAus / DecisionTransformerInterpretability
  Interpreting how transformers simulate agents performing RL tasks
  ☆80Updated last year
• callummcdougall / sae_visualizer
  ☆26Updated last year
• HazyResearch / zoology
  Understand and test language model architectures on synthetic tasks.
  ☆192Updated last month
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆111Updated 4 months ago
• redwoodresearch / interp
  Redwood Research's transformer interpretability tools
  ☆14Updated 3 years ago
• wesg52 / universal-neurons
  Universal Neurons in GPT2 Language Models
  ☆27Updated 10 months ago
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆33Updated last year