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
• ejmichaud / grokking-squared
  ☆25Updated 2 years ago
• teddykoker / grokking
  PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆36Updated 3 years ago
• KindXiaoming / Omnigrok
  Omnigrok: Grokking Beyond Algorithmic Data
  ☆56Updated 2 years ago
• locuslab / edge-of-stability
  ☆67Updated 5 months ago
• edwardjhu / TP4
  Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)
  ☆62Updated 4 years ago
• lee-ny / teaching_arithmetic
  ☆80Updated last year
• ethancaballero / broken_neural_scaling_laws
  Code Release for "Broken Neural Scaling Laws" (BNSL) paper
  ☆58Updated last year
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆48Updated last year
• epfml / llm-baselines
  nanoGPT-like codebase for LLM training
  ☆94Updated last month
• Silent-Zebra / twisted-smc-lm
  ☆28Updated last month
• GFNOrg / gfn-lm-tuning
  ☆177Updated last year
• TomFrederik / unseal
  Mechanistic Interpretability for Transformer Models
  ☆50Updated 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
• aks2203 / deep-thinking
  A centralized place for deep thinking code and experiments
  ☆84Updated last year
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆136Updated last year
• ApolloResearch / e2e_sae
  Sparse Autoencoder Training Library
  ☆49Updated 2 weeks ago
• krandiash / quinine
  A library to create and manage configuration files, especially for machine learning projects.
  ☆78Updated 3 years ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• timgaripov / compositional-sculpting
  Code for paper "Compositional Sculpting of Iterative Generative Processes"
  ☆21Updated last year
• likenneth / othello_world
  Emergent world representations: Exploring a sequence model trained on a synthetic task
  ☆181Updated last year
• redwoodresearch / Easy-Transformer
  ☆114Updated 9 months ago
• wattenberg / superposition
  Code associated to papers on superposition (in ML interpretability)
  ☆28Updated 2 years ago
• callummcdougall / sae_visualizer
  ☆27Updated last year
• bilal-chughtai / rep-theory-mech-interp
  ☆26Updated 2 years ago
• google-deepmind / neural_networks_chomsky_hierarchy
  Neural Networks and the Chomsky Hierarchy
  ☆206Updated last year
• mgrankin / minGPT
  minGPT in JAX
  ☆48Updated 3 years ago
• GFNOrg / EB_GFN
  Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"
  ☆82Updated 2 years ago
• 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
• TheDuckAI / prm
  ☆12Updated 4 months ago