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.
  ☆40Updated 2 years ago
• likenneth / othello_world
  Emergent world representations: Exploring a sequence model trained on a synthetic task
  ☆197Updated 2 years ago
• GFNOrg / gfn-lm-tuning
  ☆185Updated last year
• princeton-nlp / TransformerPrograms
  [NeurIPS 2023] Learning Transformer Programs
  ☆162Updated last year
• mechanistic-interpretability-grokking / progress-measures-paper
  ☆72Updated 3 years ago
• ethancaballero / broken_neural_scaling_laws
  Code Release for "Broken Neural Scaling Laws" (BNSL) paper
  ☆59Updated 2 years ago
• google-deepmind / neural_networks_chomsky_hierarchy
  Neural Networks and the Chomsky Hierarchy
  ☆212Updated last year
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆143Updated last year
• srush / do-we-need-attention
  ☆167Updated 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…
  ☆105Updated 2 years ago
• lee-ny / teaching_arithmetic
  ☆84Updated 2 years ago
• krandiash / quinine
  A library to create and manage configuration files, especially for machine learning projects.
  ☆79Updated 3 years ago
• taufeeque9 / codebook-features
  Sparse and discrete interpretability tool for neural networks
  ☆65Updated last year
• edwardjhu / TP4
  Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)
  ☆63Updated 4 years ago
• KindXiaoming / Omnigrok
  Omnigrok: Grokking Beyond Algorithmic Data
  ☆62Updated 2 years ago
• anthropics / toy-models-of-superposition
  Notebooks accompanying Anthropic's "Toy Models of Superposition" paper
  ☆131Updated 3 years ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆135Updated last year
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆210Updated 2 years ago
• bilal-chughtai / rep-theory-mech-interp
  ☆28Updated 2 years ago
• berlino / seq_icl
  ☆53Updated last year
• epfml / llm-baselines
  nanoGPT-like codebase for LLM training
  ☆113Updated last month
• NohTow / PPL-MCTS
  Repository for the code of the "PPL-MCTS: Constrained Textual Generation Through Discriminator-Guided Decoding" paper, NAACL'22
  ☆66Updated 3 years ago
• Sea-Snell / Implicit-Language-Q-Learning
  Official code from the paper "Offline RL for Natural Language Generation with Implicit Language Q Learning"
  ☆210Updated 2 years ago
• abhishekpanigrahi1996 / transformer_in_transformer
  ☆45Updated 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"
  ☆60Updated 3 years ago
• aks2203 / deep-thinking
  A centralized place for deep thinking code and experiments
  ☆88Updated 2 years ago
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆54Updated 2 years ago
• TomFrederik / unseal
  Mechanistic Interpretability for Transformer Models
  ☆53Updated 3 years ago
• Silent-Zebra / twisted-smc-lm
  ☆31Updated 9 months 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…
  ☆40Updated 2 years ago