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"
  ☆71Updated 2 years ago
• KindXiaoming / Omnigrok
  Omnigrok: Grokking Beyond Algorithmic Data
  ☆52Updated last year
• locuslab / edge-of-stability
  ☆63Updated 2 months ago
• lee-ny / teaching_arithmetic
  ☆77Updated last year
• ExtensityAI / benchmark
  Evaluation of neuro-symbolic engines
  ☆34Updated 6 months ago
• mlfoundations / scaling
  Language models scale reliably with over-training and on downstream tasks
  ☆96Updated 10 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆82Updated last year
• ejmichaud / grokking-squared
  ☆24Updated 2 years ago
• ApolloResearch / e2e_sae
  Sparse Autoencoder Training Library
  ☆41Updated 3 months ago
• Leiay / looped_transformer
  ☆22Updated last year
• epfml / schedules-and-scaling
  Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"
  ☆70Updated 3 months ago
• bilal-chughtai / rep-theory-mech-interp
  ☆26Updated last year
• EleutherAI / elk-generalization
  Investigating the generalization behavior of LM probes trained to predict truth labels: (1) from one annotator to another, and (2) from e…
  ☆26Updated 8 months ago
• mcleish7 / arithmetic
  Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)
  ☆183Updated 8 months ago
• hughbzhang / o1_inference_scaling_laws
  Replicating O1 inference-time scaling laws
  ☆82Updated 2 months ago
• gregorbachmann / Next-Token-Failures
  ☆80Updated 11 months ago
• katiekang1998 / reasoning_generalization
  ☆26Updated last month
• formll / dog
  DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule
  ☆58Updated last year
• Silent-Zebra / twisted-smc-lm
  ☆24Updated this week
• berlino / seq_icl
  ☆51Updated 8 months ago
• AllanYangZhou / universal_neural_functional
  ☆49Updated last year
• epfml / llm-baselines
  nanoGPT-like codebase for LLM training
  ☆89Updated this week
• jacobdunefsky / transcoder_circuits
  ☆54Updated 2 months ago
• protagolabs / odyssey-math
  ☆80Updated 3 weeks ago
• UFO-101 / auto-circuit
  A library for efficient patching and automatic circuit discovery.
  ☆53Updated 2 months ago
• wesg52 / universal-neurons
  Universal Neurons in GPT2 Language Models
  ☆27Updated 8 months 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…
  ☆16Updated 2 months ago
• abhishekpanigrahi1996 / transformer_in_transformer
  ☆44Updated last year
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆94Updated 2 months ago
• GFNOrg / gfn-lm-tuning
  ☆171Updated last year