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"
  ☆78Updated 2 years ago
• KindXiaoming / Omnigrok
  Omnigrok: Grokking Beyond Algorithmic Data
  ☆58Updated 2 years ago
• epfml / llm-baselines
  nanoGPT-like codebase for LLM training
  ☆94Updated 3 weeks ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆81Updated last year
• mlfoundations / scaling
  Language models scale reliably with over-training and on downstream tasks
  ☆97Updated last year
• lee-ny / teaching_arithmetic
  ☆82Updated last year
• abhishekpanigrahi1996 / transformer_in_transformer
  ☆45Updated last year
• ExtensityAI / benchmark
  Evaluation of neuro-symbolic engines
  ☆35Updated 10 months ago
• Silent-Zebra / twisted-smc-lm
  ☆29Updated 2 months ago
• locuslab / edge-of-stability
  ☆67Updated 5 months ago
• ejmichaud / grokking-squared
  ☆26Updated 2 years ago
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆128Updated 3 weeks ago
• callummcdougall / sae_visualizer
  ☆28Updated last year
• mechanistic-interpretability-grokking / progress-measures-paper
  ☆66Updated 2 years ago
• ApolloResearch / e2e_sae
  Sparse Autoencoder Training Library
  ☆52Updated last month
• jopetty / word-problem
  Experiments on the impact of depth in transformers and SSMs.
  ☆30Updated 7 months ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆116Updated 5 months ago
• 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
• huyphan168 / PEER
  Mixture of A Million Experts
  ☆46Updated 10 months ago
• aks2203 / deep-thinking
  A centralized place for deep thinking code and experiments
  ☆84Updated last year
• bilal-chughtai / rep-theory-mech-interp
  ☆26Updated 2 years ago
• borjanG / 2023-transformers
  Codes for the paper The emergence of clusters in self-attention dynamics.
  ☆15Updated last year
• HazyResearch / zoology
  Understand and test language model architectures on synthetic tasks.
  ☆197Updated 3 months ago
• Leiay / looped_transformer
  ☆26Updated last year
• ApolloResearch / apd
  Attribution-based Parameter Decomposition
  ☆23Updated last week
• srush / mamba-primer
  ☆37Updated last year
• IdoAmos / not-from-scratch
  ☆31Updated 7 months ago
• JeanKaddour / NoTrainNoGain
  Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)
  ☆80Updated last year
• mcleish7 / arithmetic
  Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)
  ☆190Updated last year