zhuhanqing / APOLLO
APOLLO: SGD-like Memory, AdamW-level Performance
☆212Updated 2 weeks ago
Alternatives and similar repositories for APOLLO:
Users that are interested in APOLLO are comparing it to the libraries listed below
- [ICLR 2025🔥] SVD-LLM & [NAACL 2025🔥] SVD-LLM V2☆201Updated last month
- [COLM 2024] TriForce: Lossless Acceleration of Long Sequence Generation with Hierarchical Speculative Decoding☆248Updated 8 months ago
- Efficient triton implementation of Native Sparse Attention.☆142Updated 3 weeks ago
- [ICML 2025 Spotlight] ShadowKV: KV Cache in Shadows for High-Throughput Long-Context LLM Inference☆171Updated last week
- ☆126Updated 2 months ago
- An acceleration library that supports arbitrary bit-width combinatorial quantization operations☆223Updated 7 months ago
- [NeurIPS 2024] BAdam: A Memory Efficient Full Parameter Optimization Method for Large Language Models☆257Updated last month
- The official implementation of Self-Play Preference Optimization (SPPO)☆545Updated 3 months ago
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆189Updated 2 weeks ago
- [ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.☆70Updated 4 months ago
- Unified KV Cache Compression Methods for Auto-Regressive Models☆1,039Updated 4 months ago
- 🔥 A minimal training framework for scaling FLA models☆119Updated this week
- [ICLR 2025] Palu: Compressing KV-Cache with Low-Rank Projection☆106Updated 2 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆100Updated 3 weeks ago
- Low-bit optimizers for PyTorch☆128Updated last year
- XAttention: Block Sparse Attention with Antidiagonal Scoring☆142Updated last month
- Support mixed-precsion inference with vllm☆83Updated 3 months ago
- An algorithm for weight-activation quantization (W4A4, W4A8) of LLMs, supporting both static and dynamic quantization☆131Updated 3 months ago
- 🎬 3.7× faster video generation E2E 🖼️ 1.6× faster image generation E2E ⚡ ColumnSparseAttn 9.3× vs FlashAttn‑3 💨 ColumnSparseGEMM 2.5× …☆51Updated this week
- The official implementation of MARS: Unleashing the Power of Variance Reduction for Training Large Models☆631Updated last week
- MIXQ: Taming Dynamic Outliers in Mixed-Precision Quantization by Online Prediction☆88Updated 6 months ago
- The official implementation of the paper "Towards Efficient Mixture of Experts: A Holistic Study of Compression Techniques (TMLR)".☆66Updated last month
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆160Updated last month
- Efficient LLM Inference over Long Sequences☆372Updated last week
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)☆153Updated 3 weeks ago
- ☆131Updated last month
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear at…☆101Updated 10 months ago
- Pruner-Zero: Evolving Symbolic Pruning Metric from scratch for LLMs☆81Updated 5 months ago
- ☆155Updated 2 weeks ago
- SeerAttention: Learning Intrinsic Sparse Attention in Your LLMs☆97Updated last week