ironjr / grokfast
Official repository for the paper "Grokfast: Accelerated Grokking by Amplifying Slow Gradients"
☆552Updated 10 months ago
Alternatives and similar repositories for grokfast
Users that are interested in grokfast are comparing it to the libraries listed below
Sorting:
- For optimization algorithm research and development.☆513Updated this week
- Annotated version of the Mamba paper☆483Updated last year
- Muon optimizer: +>30% sample efficiency with <3% wallclock overhead☆623Updated last month
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆281Updated last month
- Efficient optimizers☆193Updated this week
- Getting crystal-like representations with harmonic loss☆183Updated last month
- ☆180Updated 5 months ago
- Effortless plugin and play Optimizer to cut model training costs by 50%. New optimizer that is 2x faster than Adam on LLMs.☆381Updated 11 months ago
- [ICLR2025 Spotlight🔥] Official Implementation of TokenFormer: Rethinking Transformer Scaling with Tokenized Model Parameters☆555Updated 3 months ago
- Repo for "Monarch Mixer: A Simple Sub-Quadratic GEMM-Based Architecture"☆550Updated 4 months ago
- A repository for log-time feedforward networks☆222Updated last year
- The AdEMAMix Optimizer: Better, Faster, Older.☆183Updated 8 months ago
- ☆94Updated 3 months ago
- Training small GPT-2 style models using Kolmogorov-Arnold networks.☆117Updated 11 months ago
- Implementation of https://srush.github.io/annotated-s4☆494Updated 2 years ago
- Cost aware hyperparameter tuning algorithm☆152Updated 10 months ago
- Open weights language model from Google DeepMind, based on Griffin.☆639Updated 2 months ago
- ☆302Updated 10 months ago
- 🧱 Modula software package☆189Updated last month
- Normalized Transformer (nGPT)☆176Updated 5 months ago
- ☆111Updated 4 months ago
- Code repository for Black Mamba☆246Updated last year
- ☆217Updated 10 months ago
- Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).☆116Updated 6 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆233Updated 2 months ago
- Scalable and Performant Data Loading☆258Updated this week
- procedural reasoning datasets☆578Updated last week
- The history files when recording human interaction while solving ARC tasks☆109Updated 2 weeks ago
- Code to train and evaluate Neural Attention Memory Models to obtain universally-applicable memory systems for transformers.☆307Updated 6 months ago
- The boundary of neural network trainability is fractal☆198Updated last year