thu-ml / ReMoE
[ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.
β65Updated 3 months ago
Alternatives and similar repositories for ReMoE:
Users that are interested in ReMoE are comparing it to the libraries listed below
- π₯ A minimal training framework for scaling FLA modelsβ82Updated this week
- β36Updated this week
- β116Updated last month
- β70Updated last week
- β63Updated last month
- XAttention: Block Sparse Attention with Antidiagonal Scoringβ102Updated this week
- Efficient triton implementation of Native Sparse Attention.β116Updated this week
- β71Updated this week
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Modelsβ205Updated 3 weeks ago
- [ACL 2024] Not All Experts are Equal: Efficient Expert Pruning and Skipping for Mixture-of-Experts Large Language Modelsβ85Updated 10 months ago
- The official implementation of the paper <MoA: Mixture of Sparse Attention for Automatic Large Language Model Compression>β121Updated 3 months ago
- [ICML 2024 Oral] This project is the official implementation of our Accurate LoRA-Finetuning Quantization of LLMs via Information Retentiβ¦β63Updated 11 months ago
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear atβ¦β99Updated 9 months ago
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Trainingβ164Updated last month
- Code for "Everybody Prune Now: Structured Pruning of LLMs with only Forward Passes"β27Updated 11 months ago
- [ICLR 2024 Spotlight] Code for the paper "Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policy"β74Updated 9 months ago
- [NeurIPS-2024] π Scaling Laws with Vocabulary: Larger Models Deserve Larger Vocabularies https://arxiv.org/abs/2407.13623β80Updated 5 months ago
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)β150Updated 3 months ago
- A sparse attention kernel supporting mix sparse patternsβ168Updated last month
- Stick-breaking attentionβ48Updated last week
- β76Updated 2 months ago
- β36Updated 6 months ago
- β74Updated 7 months ago
- Activation-aware Singular Value Decomposition for Compressing Large Language Modelsβ59Updated 5 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inferenceβ62Updated this week
- Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"β97Updated 5 months ago
- Unofficial implementation for the paper "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"β154Updated 9 months ago
- The official implementation of paper: SimLayerKV: A Simple Framework for Layer-Level KV Cache Reduction.β43Updated 5 months ago
- Low-bit optimizers for PyTorchβ125Updated last year
- Official Implementation of SLEB: Streamlining LLMs through Redundancy Verification and Elimination of Transformer Blocksβ35Updated last month