zhuhanqing / APOLLOLinks
APOLLO: SGD-like Memory, AdamW-level Performance; MLSys'25 Oustanding Paper Honorable Mention
☆255Updated 4 months ago
Alternatives and similar repositories for APOLLO
Users that are interested in APOLLO are comparing it to the libraries listed below
Sorting:
- [COLM 2024] TriForce: Lossless Acceleration of Long Sequence Generation with Hierarchical Speculative Decoding☆264Updated last year
- [ICLR 2025🔥] SVD-LLM & [NAACL 2025🔥] SVD-LLM V2☆247Updated 3 weeks ago
- Low-bit optimizers for PyTorch☆131Updated last year
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆185Updated 3 months ago
- ☆142Updated 7 months ago
- 🔥 A minimal training framework for scaling FLA models☆247Updated last week
- [ICML 2025 Spotlight] ShadowKV: KV Cache in Shadows for High-Throughput Long-Context LLM Inference☆256Updated 4 months ago
- [NeurIPS 2024] BAdam: A Memory Efficient Full Parameter Optimization Method for Large Language Models☆268Updated 6 months ago
- Efficient triton implementation of Native Sparse Attention.☆218Updated 3 months ago
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆239Updated last month
- ☆126Updated 3 months ago
- The evaluation framework for training-free sparse attention in LLMs☆93Updated 3 months ago
- Official Implementation of SLEB: Streamlining LLMs through Redundancy Verification and Elimination of Transformer Blocks☆37Updated 7 months ago
- [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring☆229Updated 2 months ago
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models☆229Updated 4 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆141Updated 4 months ago
- Fira: Can We Achieve Full-rank Training of LLMs Under Low-rank Constraint?☆115Updated 11 months ago
- ☆243Updated 3 months ago
- ☆150Updated 3 months ago
- SeerAttention: Learning Intrinsic Sparse Attention in Your LLMs☆152Updated last month
- Official implementation of "Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decoding"☆453Updated last week
- This repository contains the training code of ParetoQ introduced in our work "ParetoQ Scaling Laws in Extremely Low-bit LLM Quantization"☆103Updated 3 months ago
- The official implementation of Self-Play Preference Optimization (SPPO)☆580Updated 7 months ago
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)☆161Updated 5 months ago
- [ICLR 2025] Palu: Compressing KV-Cache with Low-Rank Projection☆139Updated 7 months ago
- Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793☆436Updated 4 months ago
- [ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.☆91Updated 9 months ago
- Discrete Diffusion Forcing (D2F): dLLMs Can Do Faster-Than-AR Inference☆142Updated last week
- Work in progress.☆72Updated 2 months ago
- [CoLM'25] The official implementation of the paper <MoA: Mixture of Sparse Attention for Automatic Large Language Model Compression>☆146Updated 2 months ago