zhuhanqing / APOLLOLinks
APOLLO: SGD-like Memory, AdamW-level Performance; MLSys'25 Oustanding Paper Honorable Mention
β256Updated 5 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β266Updated last year
- [ICLR 2025π₯] SVD-LLM & [NAACL 2025π₯] SVD-LLM V2β255Updated last month
- β143Updated 7 months ago
- Flash-Muon: An Efficient Implementation of Muon Optimizerβ189Updated 3 months ago
- β129Updated 4 months ago
- The evaluation framework for training-free sparse attention in LLMsβ100Updated 3 months ago
- π₯ A minimal training framework for scaling FLA modelsβ259Updated last month
- Low-bit optimizers for PyTorchβ131Updated 2 years ago
- [ICML 2025 Spotlight] ShadowKV: KV Cache in Shadows for High-Throughput Long-Context LLM Inferenceβ262Updated 5 months ago
- Efficient triton implementation of Native Sparse Attention.β229Updated 4 months ago
- β251Updated 4 months ago
- [ICLR 2025] Palu: Compressing KV-Cache with Low-Rank Projectionβ142Updated 7 months ago
- [NeurIPS 2024] BAdam: A Memory Efficient Full Parameter Optimization Method for Large Language Modelsβ272Updated 6 months ago
- Discrete Diffusion Forcing (D2F): dLLMs Can Do Faster-Than-AR Inferenceβ157Updated 2 weeks ago
- Fast and memory-efficient exact attentionβ70Updated 7 months ago
- Efficient LLM Inference over Long Sequencesβ390Updated 3 months ago
- [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoringβ234Updated 3 months ago
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Trainingβ240Updated 2 months ago
- Fira: Can We Achieve Full-rank Training of LLMs Under Low-rank Constraint?β115Updated 11 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inferenceβ142Updated 4 months ago
- A curated list of recent papers on efficient video attention for video diffusion models, including sparsification, quantization, and cachβ¦β41Updated last month
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Ruleβ312Updated 3 weeks ago
- Official Implementation of SLEB: Streamlining LLMs through Redundancy Verification and Elimination of Transformer Blocksβ37Updated 8 months ago
- Towards Economical Inference: Enabling DeepSeek's Multi-Head Latent Attention in Any Transformer-based LLMsβ191Updated last week
- [ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.β95Updated 9 months ago
- Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793β437Updated 4 months ago
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear atβ¦β102Updated last year
- Get down and dirty with FlashAttention2.0 in pytorch, plug in and play no complex CUDA kernelsβ108Updated 2 years ago
- Load compute kernels from the Hubβ293Updated last week
- An efficient implementation of the NSA (Native Sparse Attention) kernelβ119Updated 3 months ago