Sea-Snell / grokkingLinks
unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
☆79Updated 3 years ago
Alternatives and similar repositories for grokking
Users that are interested in grokking are comparing it to the libraries listed below
Sorting:
- Emergent world representations: Exploring a sequence model trained on a synthetic task☆191Updated 2 years ago
- Implementation of OpenAI's 'Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets' paper.☆39Updated 2 years ago
- ☆185Updated last year
- ☆166Updated 2 years ago
- Notebooks accompanying Anthropic's "Toy Models of Superposition" paper☆130Updated 3 years ago
- Neural Networks and the Chomsky Hierarchy☆211Updated last year
- [NeurIPS 2023] Learning Transformer Programs☆162Updated last year
- ☆84Updated 2 years ago
- Omnigrok: Grokking Beyond Algorithmic Data☆62Updated 2 years ago
- Train very large language models in Jax.☆210Updated 2 years ago
- nanoGPT-like codebase for LLM training☆111Updated 3 weeks ago
- Experiments and code to generate the GINC small-scale in-context learning dataset from "An Explanation for In-context Learning as Implici…☆106Updated 2 years ago
- Code Release for "Broken Neural Scaling Laws" (BNSL) paper☆59Updated 2 years ago
- LoRA for arbitrary JAX models and functions☆143Updated last year
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆63Updated 4 years ago
- Sparse and discrete interpretability tool for neural networks☆64Updated last year
- ☆242Updated last year
- A centralized place for deep thinking code and experiments☆87Updated 2 years ago
- Scaling scaling laws with board games.☆53Updated 2 years ago
- A MAD laboratory to improve AI architecture designs 🧪☆135Updated 11 months ago
- ☆70Updated 3 years ago
- Language-annotated Abstraction and Reasoning Corpus☆97Updated 2 years ago
- ☆27Updated 2 years ago
- ☆27Updated 2 years ago
- Inference code for LLaMA models in JAX☆120Updated last year
- ☆53Updated last year
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆174Updated 5 months ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 3 years ago
- ☆29Updated last year
- ☆259Updated 6 months ago