danielmamay / grokking
Implementation of OpenAI's 'Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets' paper.
☆36Updated last year
Alternatives and similar repositories for grokking:
Users that are interested in grokking are comparing it to the libraries listed below
- unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆78Updated 2 years ago
- Omnigrok: Grokking Beyond Algorithmic Data☆55Updated 2 years ago
- ☆79Updated last year
- ☆45Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆83Updated last year
- ☆28Updated 3 weeks ago
- ☆67Updated 4 months ago
- ☆31Updated 3 months ago
- nanoGPT-like codebase for LLM training☆94Updated 3 weeks ago
- ☆25Updated last year
- ☆51Updated 11 months ago
- Code associated to papers on superposition (in ML interpretability)☆27Updated 2 years ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆71Updated 5 months ago
- ☆25Updated 2 years ago
- Evaluation of neuro-symbolic engines☆35Updated 8 months ago
- Language models scale reliably with over-training and on downstream tasks☆96Updated last year
- Code and Data Repo for the CoNLL Paper -- Future Lens: Anticipating Subsequent Tokens from a Single Hidden State☆18Updated last year
- ☆49Updated last year
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆105Updated 5 months ago
- Repository for NPHardEval, a quantified-dynamic benchmark of LLMs☆53Updated last year
- ☆31Updated 6 months ago
- A MAD laboratory to improve AI architecture designs 🧪☆111Updated 4 months ago
- Replicating O1 inference-time scaling laws☆83Updated 4 months ago
- ☆83Updated last year
- Sparse Autoencoder Training Library☆48Updated 5 months ago
- ☆31Updated 3 months ago
- Official repository for our paper, Transformers Learn Higher-Order Optimization Methods for In-Context Learning: A Study with Linear Mode…☆16Updated 5 months ago
- Machine Learning eXperiment Utilities☆46Updated 10 months ago
- ☆52Updated 6 months ago
- [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆47Updated 2 months ago