danielmamay / grokkingLinks
Implementation of OpenAI's 'Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets' paper.
☆39Updated 2 years ago
Alternatives and similar repositories for grokking
Users that are interested in grokking are comparing it to the libraries listed below
Sorting:
- unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆79Updated 3 years ago
- Omnigrok: Grokking Beyond Algorithmic Data☆62Updated 2 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆88Updated last year
- ☆45Updated last year
- nanoGPT-like codebase for LLM training☆107Updated 4 months ago
- Evaluation of neuro-symbolic engines☆39Updated last year
- ☆166Updated 2 years ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆84Updated 10 months ago
- Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)☆192Updated last year
- A centralized place for deep thinking code and experiments☆86Updated 2 years ago
- Sparse and discrete interpretability tool for neural networks☆63Updated last year
- ☆82Updated last year
- Official repository for our paper, Transformers Learn Higher-Order Optimization Methods for In-Context Learning: A Study with Linear Mode…☆18Updated 10 months ago
- ☆58Updated 11 months ago
- Code for the paper "The Impact of Positional Encoding on Length Generalization in Transformers", NeurIPS 2023☆136Updated last year
- ☆33Updated 8 months ago
- Language models scale reliably with over-training and on downstream tasks☆99Updated last year
- Universal Neurons in GPT2 Language Models☆30Updated last year
- ☆53Updated last year
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆162Updated 2 months ago
- ☆83Updated 2 years ago
- ☆39Updated 3 weeks ago
- Mixture of A Million Experts☆47Updated last year
- ☆31Updated 5 months ago
- ☆11Updated last year
- Sparse Autoencoder Training Library☆54Updated 4 months ago
- Code for reproducing our paper "Not All Language Model Features Are Linear"☆80Updated 9 months ago
- ☆100Updated last year
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule☆63Updated 2 years ago
- A MAD laboratory to improve AI architecture designs 🧪☆129Updated 9 months ago