Alternatives and similar repositories for Bi-Mamba2:

Users that are interested in Bi-Mamba2 are comparing it to the libraries listed below

• goombalab / hydra
  Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"
  ☆126Updated 2 months ago
• AmeenAli / HiddenMambaAttn
  Official PyTorch Implementation of "The Hidden Attention of Mamba Models"
  ☆218Updated 10 months ago
• MzeroMiko / mamba-mini
  An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivatio…
  ☆81Updated last year
• kyegomez / Griffin
  Implementation of Griffin from the paper: "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"
  ☆52Updated last week
• tommyip / mamba2-minimal
  Minimal Mamba-2 implementation in PyTorch
  ☆181Updated 9 months ago
• Adamdad / rational_kat_cu
  ☆57Updated 2 months ago
• jacklishufan / Mamba-ND
  Ofiicial Implementation for Mamba-ND: Selective State Space Modeling for Multi-Dimensional Data
  ☆58Updated 9 months ago
• pengzhangzhi / Awesome-Mamba
  Awesome list of papers that extend Mamba to various applications.
  ☆132Updated last week
• badripatro / simba
  Simba
  ☆206Updated last year
• YuHengsss / VSSD
  Introduce Mamba2 to Vision.
  ☆126Updated 7 months ago
• WailordHe / DenseSSM
  A repository for DenseSSMs
  ☆87Updated last year
• Itamarzimm / UnifiedImplicitAttnRepr
  [ICLR 2025] Official Code Release for Explaining Modern Gated-Linear RNNs via a Unified Implicit Attention Formulation
  ☆42Updated last month
• OliverRensu / ARM
  This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision
  ☆72Updated 9 months ago
• NVlabs / ConvSSM
  ☆64Updated 5 months ago
• goombalab / phi-mamba
  Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Mode…
  ☆103Updated 7 months ago
• assafbk / DeciMamba
  DeciMamba: Exploring the Length Extrapolation Potential of Mamba (ICLR 2025)
  ☆25Updated this week
• badripatro / mamba360
  State Space Models
  ☆67Updated 11 months ago
• ethanbar11 / ssm_2d
  More dimensions = More fun
  ☆21Updated 8 months ago
• Dao-AILab / causal-conv1d
  Causal depthwise conv1d in CUDA, with a PyTorch interface
  ☆427Updated 4 months ago
• MambaMixer / M2
  ☆48Updated last year
• kyegomez / SwitchTransformers
  Implementation of Switch Transformers from the paper: "Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficien…
  ☆96Updated last week
• LeapLabTHU / MLLA
  Official repository of MLLA (NeurIPS 2024)
  ☆309Updated 4 months ago
• kyegomez / MambaTransformer
  Integrating Mamba/SSMs with Transformer for Enhanced Long Context and High-Quality Sequence Modeling
  ☆191Updated last week
• nanowell / Differential-Transformer-PyTorch
  PyTorch implementation of the Differential-Transformer architecture for sequence modeling, specifically tailored as a decoder-only model …
  ☆57Updated 5 months ago
• wangf3014 / Mamba-Reg
  ☆64Updated last month
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆148Updated 3 weeks ago
• EasonXiao-888 / MambaTree
  [NeurIPS2024 Spotlight] The official implementation of MambaTree: Tree Topology is All You Need in State Space Model
  ☆92Updated 10 months ago
• naver-ai / rope-vit
  [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"
  ☆305Updated 3 months ago
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆82Updated last month
• Amshaker / GroupMamba
  GroupMamba: Parameter-Efficient and Accurate Group Visual State Space Model [CVPR -2025]
  ☆89Updated 3 weeks ago