Alternatives and similar repositories for grokking

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

• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆79Updated 3 years ago
• epfml / llm-baselines
  nanoGPT-like codebase for LLM training
  ☆109Updated last week
• abhishekpanigrahi1996 / transformer_in_transformer
  ☆45Updated 2 years ago
• mlfoundations / scaling
  Language models scale reliably with over-training and on downstream tasks
  ☆100Updated last year
• mcleish7 / arithmetic
  Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)
  ☆193Updated last year
• taufeeque9 / codebook-features
  Sparse and discrete interpretability tool for neural networks
  ☆64Updated last year
• KindXiaoming / Omnigrok
  Omnigrok: Grokking Beyond Algorithmic Data
  ☆62Updated 2 years ago
• aks2203 / deep-thinking
  A centralized place for deep thinking code and experiments
  ☆87Updated 2 years ago
• 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…
  ☆19Updated 11 months ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆133Updated 10 months ago
• epfml / schedules-and-scaling
  Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"
  ☆85Updated last year
• McGill-NLP / length-generalization
  Code for the paper "The Impact of Positional Encoding on Length Generalization in Transformers", NeurIPS 2023
  ☆137Updated last year
• epfml / DenseFormer
  ☆81Updated last year
• srush / do-we-need-attention
  ☆166Updated 2 years ago
• katiekang1998 / reasoning_generalization
  ☆33Updated 10 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆89Updated last year
• GFNOrg / gfn-lm-tuning
  ☆185Updated last year
• Silent-Zebra / twisted-smc-lm
  ☆32Updated 7 months ago
• apple / ml-planner
  ☆56Updated last year
• allenai / fluid-benchmarking
  Fluid Language Model Benchmarking
  ☆20Updated 2 months ago
• wesg52 / universal-neurons
  Universal Neurons in GPT2 Language Models
  ☆31Updated last year
• ExtensityAI / benchmark
  Evaluation of neuro-symbolic engines
  ☆39Updated last year
• zhangir-azerbayev / MetaMath
  ☆11Updated 2 years ago
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆172Updated 4 months ago
• ApolloResearch / e2e_sae
  Sparse Autoencoder Training Library
  ☆55Updated 6 months ago
• KoyenaPal / future-lens
  Code and Data Repo for the CoNLL Paper -- Future Lens: Anticipating Subsequent Tokens from a Single Hidden State
  ☆20Updated 3 weeks ago
• lucidrains / pause-transformer
  Yet another random morning idea to be quickly tried and architecture shared if it works; to allow the transformer to pause for any amount…
  ☆52Updated 2 years ago
• berlino / seq_icl
  ☆53Updated last year
• lee-ny / teaching_arithmetic
  ☆83Updated 2 years ago
• OSU-NLP-Group / GrokkedTransformer
  Code for NeurIPS'24 paper 'Grokked Transformers are Implicit Reasoners: A Mechanistic Journey to the Edge of Generalization'
  ☆233Updated 3 months ago