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
  ☆58Updated 2 years ago
• lee-ny / teaching_arithmetic
  ☆82Updated last year
• TomFrederik / unseal
  Mechanistic Interpretability for Transformer Models
  ☆51Updated 3 years ago
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆49Updated last year
• edwardjhu / TP4
  Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)
  ☆62Updated 4 years ago
• ejmichaud / grokking-squared
  ☆26Updated 2 years ago
• redwoodresearch / Easy-Transformer
  ☆117Updated 10 months ago
• redwoodresearch / interp
  Redwood Research's transformer interpretability tools
  ☆15Updated 3 years ago
• mechanistic-interpretability-grokking / progress-measures-paper
  ☆66Updated 2 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…
  ☆106Updated last year
• teddykoker / grokking
  PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆37Updated 3 years ago
• ethancaballero / broken_neural_scaling_laws
  Code Release for "Broken Neural Scaling Laws" (BNSL) paper
  ☆58Updated last year
• berlino / seq_icl
  ☆52Updated last year
• epfml / schedules-and-scaling
  Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"
  ☆73Updated 7 months ago
• locuslab / edge-of-stability
  ☆67Updated 6 months ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆136Updated last year
• EdinburghNLP / torch-adaptive-imle
  ☆34Updated 6 months ago
• bilal-chughtai / rep-theory-mech-interp
  ☆26Updated 2 years ago
• GFNOrg / gfn-lm-tuning
  ☆179Updated last year
• timaeus-research / devinterp
  Tools for studying developmental interpretability in neural networks.
  ☆91Updated 4 months ago
• princeton-nlp / TransformerPrograms
  [NeurIPS 2023] Learning Transformer Programs
  ☆161Updated last year
• Silent-Zebra / twisted-smc-lm
  ☆29Updated 2 months ago
• TheDuckAI / prm
  ☆12Updated 4 months ago
• ApolloResearch / apd
  Attribution-based Parameter Decomposition
  ☆24Updated this week
• ApolloResearch / e2e_sae
  Sparse Autoencoder Training Library
  ☆52Updated last month
• callummcdougall / sae_visualizer
  ☆28Updated last year
• DeqingFu / transformers-icl-second-order
  Official repository for our paper, Transformers Learn Higher-Order Optimization Methods for In-Context Learning: A Study with Linear Mode…
  ☆16Updated 6 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆81Updated last year
• wattenberg / superposition
  Code associated to papers on superposition (in ML interpretability)
  ☆28Updated 2 years ago