Alternatives and similar repositories for LongMamba

Users that are interested in LongMamba are comparing it to the libraries listed below

• HazyResearch / based
  Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"
  ☆235Updated 2 weeks ago
• HazyResearch / zoology
  Understand and test language model architectures on synthetic tasks.
  ☆217Updated 2 weeks ago
• goombalab / phi-mamba
  Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Mode…
  ☆108Updated 9 months ago
• jxiw / MambaInLlama
  [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models
  ☆221Updated last month
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆120Updated 6 months ago
• NVIDIA / ngpt
  Normalized Transformer (nGPT)
  ☆183Updated 7 months ago
• lucidrains / PEER-pytorch
  Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind
  ☆127Updated 10 months ago
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆173Updated 3 months ago
• HazyResearch / lolcats
  Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"
  ☆239Updated 4 months ago
• llm-random / llm-random
  ☆190Updated this week
• fla-org / flame
  🔥 A minimal training framework for scaling FLA models
  ☆170Updated last week
• epfml / dynamic-sparse-flash-attention
  ☆147Updated 2 years ago
• lucidrains / nGPT-pytorch
  Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI
  ☆284Updated 3 weeks ago
• astramind-ai / Mixture-of-depths
  Unofficial implementation for the paper "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"
  ☆163Updated last year
• lucidrains / coconut-pytorch
  Implementation of 🥥 Coconut, Chain of Continuous Thought, in Pytorch
  ☆174Updated 5 months ago
• schwartz-lab-NLP / TOVA
  Token Omission Via Attention
  ☆127Updated 8 months ago
• shawntan / stickbreaking-attention
  Stick-breaking attention
  ☆57Updated last week
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆134Updated this week
• thu-ml / ReMoE
  [ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.
  ☆81Updated 6 months ago
• shawntan / scattermoe
  Triton-based implementation of Sparse Mixture of Experts.
  ☆219Updated 6 months ago
• itsnamgyu / block-transformer
  Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)
  ☆158Updated 2 months ago
• huyphan168 / PEER
  Mixture of A Million Experts
  ☆46Updated 10 months ago
• RobertCsordas / moe_attention
  Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"
  ☆98Updated 8 months ago
• berlino / gated_linear_attention
  ☆105Updated last year
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆126Updated 6 months ago
• zhixuan-lin / forgetting-transformer
  [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"
  ☆108Updated last month
• sustcsonglin / linear-attention-and-beyond-slides
  ☆76Updated 3 months ago
• XunhaoLai / native-sparse-attention-triton
  Efficient triton implementation of Native Sparse Attention.
  ☆168Updated last month
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆182Updated 10 months ago
• alexzhang13 / flashattention2-custom-mask
  Triton implementation of FlashAttention2 that adds Custom Masks.
  ☆120Updated 10 months ago