bloc97 / DeMo
DeMo: Decoupled Momentum Optimization
☆185Updated 4 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☆80Updated 4 months ago
- Focused on fast experimentation and simplicity☆71Updated 3 months ago
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆95Updated last month
- Normalized Transformer (nGPT)☆167Updated 4 months ago
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆230Updated 2 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆103Updated 4 months ago
- research impl of Native Sparse Attention (2502.11089)☆53Updated last month
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆125Updated 4 months ago
- ☆92Updated 2 months ago
- PyTorch implementation of models from the Zamba2 series.☆179Updated 2 months ago
- Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradients.☆195Updated 8 months ago
- WIP☆93Updated 8 months ago
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆123Updated 7 months ago
- smolLM with Entropix sampler on pytorch☆151Updated 5 months ago
- ☆76Updated 9 months ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆60Updated last week
- An implementation of PSGD Kron second-order optimizer for PyTorch☆86Updated 2 weeks ago
- A MAD laboratory to improve AI architecture designs 🧪☆109Updated 3 months ago
- Collection of autoregressive model implementation☆85Updated 2 months ago
- look how they massacred my boy☆63Updated 5 months ago
- ☆166Updated 2 months ago
- ☆107Updated 3 months ago
- Efficient optimizers☆188Updated this week
- When it comes to optimizers, it's always better to be safe than sorry☆217Updated 2 weeks ago
- OpenCoconut implements a latent reasoning paradigm where we generate thoughts before decoding.☆170Updated 2 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆123Updated 11 months ago
- Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers☆90Updated 8 months ago
- ☆129Updated 7 months ago
- ☆97Updated 6 months ago
- ☆43Updated last year