zhuhanqing / APOLLOLinks
APOLLO: SGD-like Memory, AdamW-level Performance; MLSys'25 Oustanding Paper Honorable Mention
☆267Updated last month
Alternatives and similar repositories for APOLLO
Users that are interested in APOLLO are comparing it to the libraries listed below
Sorting:
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆225Updated 6 months ago
- [COLM 2024] TriForce: Lossless Acceleration of Long Sequence Generation with Hierarchical Speculative Decoding☆274Updated last year
- [ICLR 2025🔥] SVD-LLM & [NAACL 2025🔥] SVD-LLM V2☆270Updated 4 months ago
- ☆157Updated 10 months ago
- 🔥 A minimal training framework for scaling FLA models☆333Updated last month
- [NeurIPS 2024] BAdam: A Memory Efficient Full Parameter Optimization Method for Large Language Models☆281Updated 9 months ago
- ☆133Updated 7 months ago
- Low-bit optimizers for PyTorch☆137Updated 2 years ago
- The evaluation framework for training-free sparse attention in LLMs☆108Updated 2 months ago
- [ICML 2024] Official Implementation of SLEB: Streamlining LLMs through Redundancy Verification and Elimination of Transformer Blocks☆38Updated 11 months ago
- Fast and memory-efficient exact attention☆75Updated 10 months ago
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆249Updated 11 months ago
- Towards Economical Inference: Enabling DeepSeek's Multi-Head Latent Attention in Any Transformer-based LLMs☆198Updated last month
- [ICML 2025 Spotlight] ShadowKV: KV Cache in Shadows for High-Throughput Long-Context LLM Inference☆278Updated 8 months ago
- Fira: Can We Achieve Full-rank Training of LLMs Under Low-rank Constraint?☆119Updated last year
- [ICLR 2025] Palu: Compressing KV-Cache with Low-Rank Projection☆151Updated 10 months ago
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models☆233Updated 2 months ago
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆256Updated 4 months ago
- ☆49Updated last year
- Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793☆448Updated 7 months ago
- ☆102Updated 10 months ago
- ☆85Updated last month
- ☆263Updated 7 months ago
- Unofficial implementation for the paper "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"☆176Updated last year
- Code for studying the super weight in LLM☆121Updated last year
- [ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.☆104Updated last year
- ☆235Updated last year
- Official implementation for Training LLMs with MXFP4☆116Updated 8 months ago
- Triton-based implementation of Sparse Mixture of Experts.☆259Updated 3 months ago
- Efficient triton implementation of Native Sparse Attention.☆257Updated 7 months ago