Alternatives and similar repositories for speculative_mamba

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

• TUDa-HWAI / Basis_Sharing
  ☆14Updated 11 months ago
• thu-ml / Adaptive-Sparse-Trainer
  Official implementation for "Pruning Large Language Models with Semi-Structural Adaptive Sparse Training" (AAAI 2025)
  ☆12Updated 2 months ago
• krafton-ai / lexico
  KV cache compression via sparse coding
  ☆12Updated 3 months ago
• mengxiayu / LLMSuperWeight
  Code for studying the super weight in LLM
  ☆117Updated 9 months ago
• ambisinister / mla-experiments
  Experiments on Multi-Head Latent Attention
  ☆95Updated last year
• xiayuqing0622 / flex_head_fa
  Fast and memory-efficient exact attention
  ☆69Updated 6 months ago
• martin-marek / batch-size
  📄Small Batch Size Training for Language Models
  ☆57Updated last week
• amazon-science / mxfp4-llm
  Official implementation for Training LLMs with MXFP4
  ☆79Updated 4 months ago
• shikaiqiu / compute-better-spent
  ☆57Updated 11 months ago
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆187Updated last year
• insuhan / hyper-attn
  ☆82Updated last year
• NVIDIA / ngpt
  Normalized Transformer (nGPT)
  ☆187Updated 9 months ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆127Updated 8 months ago
• apple / ml-dataset-decomposition
  Official repo of dataset-decomposition paper [NeurIPS 2024]
  ☆19Updated 7 months ago
• IST-DASLab / QuEST
  Work in progress.
  ☆72Updated 2 months ago
• apple / ml-sigmoid-attention
  ☆298Updated 4 months ago
• jxiw / MambaInLlama
  [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models
  ☆228Updated 4 months ago
• assafbk / DeciMamba
  DeciMamba: Exploring the Length Extrapolation Potential of Mamba (ICLR 2025)
  ☆30Updated 4 months ago
• nil0x9 / flash-muon
  Flash-Muon: An Efficient Implementation of Muon Optimizer
  ☆176Updated 2 months ago
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆209Updated 5 months ago
• PiotrNawrot / sparse-frontier
  The evaluation framework for training-free sparse attention in LLMs
  ☆91Updated 2 months ago
• NX-AI / flashrnn
  FlashRNN - Fast RNN Kernels with I/O Awareness
  ☆97Updated 2 months ago
• enyac-group / Quamba
  The official repository of Quamba1 [ICLR 2025] & Quamba2 [ICML 2025]
  ☆59Updated 2 months ago
• HanGuo97 / log-linear-attention
  ☆240Updated 2 months ago
• HazyResearch / zoology
  Understand and test language model architectures on synthetic tasks.
  ☆222Updated last month
• PiotrNawrot / nano-sparse-attention
  The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.
  ☆85Updated last month
• google-deepmind / asyncdiloco
  ☆46Updated last year
• goombalab / phi-mamba
  Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Mode…
  ☆114Updated 11 months ago
• UNITES-Lab / HEXA-MoE
  Official code for the paper "HEXA-MoE: Efficient and Heterogeneous-Aware MoE Acceleration with Zero Computation Redundancy"
  ☆13Updated 5 months ago
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆129Updated 9 months ago