jxiw / M1Links
M1: Towards Scalable Test-Time Compute with Mamba Reasoning Models
☆26Updated 2 weeks ago
Alternatives and similar repositories for M1
Users that are interested in M1 are comparing it to the libraries listed below
Sorting:
- ☆54Updated 3 weeks ago
- [ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.☆85Updated 7 months ago
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models☆225Updated 3 months ago
- ☆95Updated 3 months ago
- Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Mode…☆113Updated 10 months ago
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆108Updated last month
- ☆79Updated 5 months ago
- 🔥 A minimal training framework for scaling FLA models☆209Updated last month
- Implementation of the paper: "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"☆103Updated last week
- An unofficial implementation of "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"☆35Updated last year
- [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"☆118Updated last month
- ☆113Updated last month
- Code for "Reasoning to Learn from Latent Thoughts"☆114Updated 4 months ago
- [CoLM'25] The official implementation of the paper <MoA: Mixture of Sparse Attention for Automatic Large Language Model Compression>☆141Updated 3 weeks ago
- ☆123Updated 2 months ago
- Kinetics: Rethinking Test-Time Scaling Laws☆70Updated 3 weeks ago
- [ICLR 2024 Spotlight] Code for the paper "Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policy"☆88Updated last month
- ☆88Updated 2 months ago
- Some preliminary explorations of Mamba's context scaling.☆216Updated last year
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆192Updated 4 months ago
- ☆81Updated last week
- The official code implementation for paper "R2R: Efficiently Navigating Divergent Reasoning Paths with Small-Large Model Token Routing"☆43Updated last week
- [COLM 2025] Code for Paper: Learning Adaptive Parallel Reasoning with Language Models☆117Updated 3 months ago
- Activation-aware Singular Value Decomposition for Compressing Large Language Models☆74Updated 9 months ago
- Stick-breaking attention☆58Updated last month
- [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring☆213Updated 3 weeks ago
- Implementation of 🥥 Coconut, Chain of Continuous Thought, in Pytorch☆178Updated last month
- Efficient triton implementation of Native Sparse Attention.☆186Updated 2 months ago
- ☆10Updated 10 months ago
- [ICML 2024 Oral] This project is the official implementation of our Accurate LoRA-Finetuning Quantization of LLMs via Information Retenti…☆67Updated last year