Sea-Snell / grokking
unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
☆55Updated 2 years ago
Related projects: ⓘ
- Implementation of OpenAI's 'Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets' paper.☆28Updated 11 months ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆57Updated 3 years ago
- Omnigrok: Grokking Beyond Algorithmic Data☆45Updated last year
- ☆21Updated last year
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated last year
- PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆31Updated 2 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆58Updated 2 years ago
- This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…☆34Updated last year
- ☆46Updated 7 months ago
- A centralized place for deep thinking code and experiments☆73Updated last year
- Language-annotated Abstraction and Reasoning Corpus☆76Updated last year
- Code Release for "Broken Neural Scaling Laws" (BNSL) paper☆57Updated 10 months ago
- ☆56Updated 3 years ago
- ☆50Updated last year
- ☆27Updated this week
- Official implementation of "Multi-scale Feature Learning Dynamics: Insights for Double Descent".☆16Updated 2 years ago
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆73Updated last year
- Jax/Flax rewrite of Karpathy's nanoGPT☆46Updated last year
- Scaling scaling laws with board games.☆36Updated last year
- LoRA for arbitrary JAX models and functions☆127Updated 6 months ago
- Sparse and discrete interpretability tool for neural networks☆51Updated 7 months ago
- ☆28Updated this week
- Interpreting how transformers simulate agents performing RL tasks☆62Updated 10 months ago
- ☆68Updated 3 weeks ago
- ☆33Updated 3 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆74Updated 7 months ago
- ☆99Updated last month
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆42Updated last year
- ☆68Updated 7 months ago
- Influence Functions with (Eigenvalue-corrected) Kronecker-Factored Approximate Curvature☆93Updated last month