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"
  ☆138Updated 4 months ago
• AmeenAli / HiddenMambaAttn
  Official PyTorch Implementation of "The Hidden Attention of Mamba Models"
  ☆223Updated last year
• MzeroMiko / mamba-mini
  An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivatio…
  ☆91Updated last year
• jacklishufan / Mamba-ND
  Ofiicial Implementation for Mamba-ND: Selective State Space Modeling for Multi-Dimensional Data
  ☆60Updated 11 months ago
• Adamdad / rational_kat_cu
  ☆63Updated 4 months ago
• badripatro / simba
  Simba
  ☆208Updated last year
• tommyip / mamba2-minimal
  Minimal Mamba-2 implementation in PyTorch
  ☆202Updated last year
• pengzhangzhi / Awesome-Mamba
  Awesome list of papers that extend Mamba to various applications.
  ☆133Updated 2 weeks ago
• kyegomez / Griffin
  Implementation of Griffin from the paper: "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"
  ☆55Updated 2 months ago
• assafbk / DeciMamba
  DeciMamba: Exploring the Length Extrapolation Potential of Mamba (ICLR 2025)
  ☆28Updated 2 months ago
• YuHengsss / VSSD
  Introduce Mamba2 to Vision.
  ☆132Updated 10 months ago
• NVlabs / ConvSSM
  ☆65Updated 8 months ago
• naver-ai / rope-vit
  [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"
  ☆335Updated 6 months ago
• LeapLabTHU / MLLA
  Official repository of MLLA (NeurIPS 2024)
  ☆330Updated 6 months ago
• WailordHe / DenseSSM
  A repository for DenseSSMs
  ☆87Updated last year
• badripatro / mamba360
  State Space Models
  ☆67Updated last year
• fla-org / flash-bidirectional-linear-attention
  Triton implement of bi-directional (non-causal) linear attention
  ☆50Updated 4 months ago
• fkodom / yet-another-retnet
  A simple but robust PyTorch implementation of RetNet from "Retentive Network: A Successor to Transformer for Large Language Models" (http…
  ☆106Updated last year
• qhfan / RALA
  [CVPR2025] Breaking the Low-Rank Dilemma of Linear Attention
  ☆22Updated 3 months ago
• OliverRensu / ARM
  [ICLR2025] This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision
  ☆80Updated 3 weeks ago
• ethanbar11 / ssm_2d
  More dimensions = More fun
  ☆22Updated 10 months ago
• kyegomez / MambaTransformer
  Integrating Mamba/SSMs with Transformer for Enhanced Long Context and High-Quality Sequence Modeling
  ☆196Updated 2 months ago
• Dao-AILab / causal-conv1d
  Causal depthwise conv1d in CUDA, with a PyTorch interface
  ☆494Updated 3 weeks ago
• lucidrains / agent-attention-pytorch
  Implementation of Agent Attention in Pytorch
  ☆90Updated 11 months ago
• nanowell / Differential-Transformer-PyTorch
  PyTorch implementation of the Differential-Transformer architecture for sequence modeling, specifically tailored as a decoder-only model …
  ☆70Updated 7 months ago
• Itamarzimm / UnifiedImplicitAttnRepr
  [ICLR 2025] Official Code Release for Explaining Modern Gated-Linear RNNs via a Unified Implicit Attention Formulation
  ☆42Updated 3 months ago
• kyegomez / MoE-Mamba
  Implementation of MoE Mamba from the paper: "MoE-Mamba: Efficient Selective State Space Models with Mixture of Experts" in Pytorch and Ze…
  ☆107Updated 2 months ago
• hkproj / mamba-notes
  Notes on the Mamba and the S4 model (Mamba: Linear-Time Sequence Modeling with Selective State Spaces)
  ☆169Updated last year
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆173Updated 3 months ago
• BICLab / MetaLA
  Offical implementation of "MetaLA: Unified Optimal Linear Approximation to Softmax Attention Map" (NeurIPS2024 Oral)
  ☆25Updated 5 months ago