bloc97 / DeMo
DeMo: Decoupled Momentum Optimization
☆180Updated 2 months ago
Alternatives and similar repositories for DeMo:
Users that are interested in DeMo are comparing it to the libraries listed below
- supporting pytorch FSDP for optimizers☆76Updated 2 months ago
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆215Updated 3 weeks ago
- PyTorch implementation of models from the Zamba2 series.☆176Updated 3 weeks ago
- Normalized Transformer (nGPT)☆152Updated 3 months ago
- Muon optimizer: +~30% sample efficiency with <3% wallclock overhead☆253Updated last week
- Focused on fast experimentation and simplicity☆65Updated last month
- smolLM with Entropix sampler on pytorch☆150Updated 3 months ago
- ☆82Updated last month
- Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradients.☆191Updated 7 months ago
- look how they massacred my boy☆63Updated 4 months ago
- Efficient optimizers☆169Updated this week
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆95Updated 3 months ago
- ☆100Updated last month
- ☆96Updated 4 months ago
- ☆75Updated 7 months ago
- Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers☆82Updated 7 months ago
- A MAD laboratory to improve AI architecture designs 🧪☆102Updated 2 months ago
- An efficent implementation of the method proposed in "The Era of 1-bit LLMs"☆154Updated 4 months ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆95Updated last month
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆118Updated 5 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆116Updated 2 months ago
- ☆53Updated last year
- WIP☆93Updated 6 months ago
- Simple Transformer in Jax☆136Updated 7 months ago
- ☆49Updated 11 months ago
- ☆158Updated 2 months ago