Alternatives and similar repositories for ssm_2d

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

• philippe-eecs / small-vision
  A big_vision inspired repo that implements a generic Auto-Encoder class capable in representation learning and generative modeling.
  ☆34Updated last year
• OliverRensu / ARM
  [ICLR2025] This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision
  ☆87Updated 6 months ago
• AmeenAli / HiddenMambaAttn
  Official PyTorch Implementation of "The Hidden Attention of Mamba Models"
  ☆231Updated last month
• TomerRonen34 / mixed-resolution-vit
  ☆54Updated 2 years ago
• goombalab / hydra
  Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"
  ☆166Updated 10 months ago
• TonyLianLong / CrossMAE
  Official Implementation of the CrossMAE paper: Rethinking Patch Dependence for Masked Autoencoders
  ☆126Updated 8 months ago
• alhojel / visual_task_vectors
  ☆40Updated last year
• OliverRensu / D-iGPT
  [ICML 2024] This repository includes the official implementation of our paper "Rejuvenating image-GPT as Strong Visual Representation Lea…
  ☆98Updated last year
• Optimization-AI / FastCLIP
  Distributed Optimization Infra for learning CLIP models
  ☆27Updated last year
• nickjiang2378 / test-time-registers
  [NeurIPS '25 Spotlight] Official Pytorch implementation of "Vision Transformers Don't Need Trained Registers"
  ☆145Updated 2 months ago
• s-sahoo / MuLAN
  [NeurIPS 2024, spotlight] Multivariate Learned Adaptive Noise for Diffusion Models
  ☆30Updated 11 months ago
• CompVis / discrete-interpolants
  The official implementation of "[MASK] is All You Need"
  ☆125Updated 4 months ago
• pengzhangzhi / Awesome-Mamba
  Awesome list of papers that extend Mamba to various applications.
  ☆139Updated 5 months ago
• UCDvision / NOLA
  Code for NOLA, an implementation of "nola: Compressing LoRA using Linear Combination of Random Basis"
  ☆57Updated last year
• MambaMixer / M2
  ☆48Updated last year
• hammoudhasan / DiffCLIP
  Official Implementation of DiffCLIP: Differential Attention Meets CLIP
  ☆48Updated 8 months ago
• hyperevolnet / Terminator
  The official repository for HyperZ⋅Z⋅W Operator Connects Slow-Fast Networks for Full Context Interaction.
  ☆43Updated 8 months ago
• HelmholtzAI-FZJ / flex_gen
  ☆19Updated 10 months ago
• caojiaolong / Awesome-Mamba
  Collect papers about Mamba (a selective state space model).
  ☆14Updated last year
• minhoooo1 / CatMAE
  CatMAE
  ☆14Updated last year
• facebookresearch / RCDM
  Visualizing representations with diffusion based conditional generative model.
  ☆103Updated 2 years ago
• MarkXCloud / CSpD
  The official repo of continuous speculative decoding
  ☆30Updated 8 months ago
• Hprairie / Bi-Mamba2
  A Triton Kernel for incorporating Bi-Directionality in Mamba2
  ☆75Updated 11 months ago
• ChenDelong1999 / subobjects
  Official repository of paper "Subobject-level Image Tokenization" (ICML-25)
  ☆90Updated 5 months ago
• Adamdad / Repfusion
  ☆57Updated 2 years ago
• bwconrad / soft-moe
  PyTorch implementation of "From Sparse to Soft Mixtures of Experts"
  ☆66Updated 2 years ago
• lucidrains / LVMAE-pytorch
  Implementation of the proposed LVMAE, from the paper, Extending Video Masked Autoencoders to 128 frames, in Pytorch
  ☆55Updated last year
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆107Updated last year
• ExplainableML / fomo_in_flux
  Code and benchmark for the paper: "A Practitioner's Guide to Continual Multimodal Pretraining" [NeurIPS'24]
  ☆60Updated 11 months ago
• james-oldfield / muMoE
  [NeurIPS'24] Multilinear Mixture of Experts: Scalable Expert Specialization through Factorization
  ☆37Updated last year