Alternatives and similar repositories for ssm_2d

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

• AmeenAli / HiddenMambaAttn
  Official PyTorch Implementation of "The Hidden Attention of Mamba Models"
  ☆231Updated 3 months ago
• OliverRensu / ARM
  [ICLR2025] This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision
  ☆90Updated 8 months ago
• TonyLianLong / CrossMAE
  Official Implementation of the CrossMAE paper: Rethinking Patch Dependence for Masked Autoencoders
  ☆131Updated 10 months ago
• alhojel / visual_task_vectors
  ☆40Updated last year
• hammoudhasan / DiffCLIP
  Official Implementation of DiffCLIP: Differential Attention Meets CLIP
  ☆51Updated 11 months ago
• 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
• AILab-CVC / M2PT
  [CVPR 2024] Multimodal Pathway: Improve Transformers with Irrelevant Data from Other Modalities
  ☆101Updated last year
• 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
• s-sahoo / MuLAN
  [NeurIPS 2024, spotlight] Multivariate Learned Adaptive Noise for Diffusion Models
  ☆31Updated last year
• MambaMixer / M2
  ☆48Updated last year
• goombalab / hydra
  Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"
  ☆170Updated last year
• PKU-ML / non_neg
  Official Code for ICLR 2024 Paper: Non-negative Contrastive Learning
  ☆47Updated last year
• UCDvision / NOLA
  Code for NOLA, an implementation of "nola: Compressing LoRA using Linear Combination of Random Basis"
  ☆57Updated last year
• minhoooo1 / CatMAE
  CatMAE
  ☆14Updated 2 years ago
• ChenDelong1999 / subobjects
  Official repository of paper "Subobject-level Image Tokenization" (ICML-25)
  ☆92Updated 7 months ago
• pengzhangzhi / Awesome-Mamba
  Awesome list of papers that extend Mamba to various applications.
  ☆138Updated 8 months ago
• google-deepmind / tips
  TIPS (ICLR'25): Text-Image Pretraining with Spatial Awareness
  ☆117Updated 10 months ago
• nickjiang2378 / test-time-registers
  [NeurIPS '25 Spotlight] Official Pytorch implementation of "Vision Transformers Don't Need Trained Registers"
  ☆172Updated 4 months ago
• TomerRonen34 / mixed-resolution-vit
  ☆56Updated 2 years ago
• facebookresearch / RCDM
  Visualizing representations with diffusion based conditional generative model.
  ☆106Updated 2 years ago
• caojiaolong / Awesome-Mamba
  Collect papers about Mamba (a selective state space model).
  ☆14Updated last year
• james-oldfield / muMoE
  [NeurIPS'24] Multilinear Mixture of Experts: Scalable Expert Specialization through Factorization
  ☆39Updated last year
• ExplainableML / fomo_in_flux
  Code and benchmark for the paper: "A Practitioner's Guide to Continual Multimodal Pretraining" [NeurIPS'24]
  ☆61Updated last year
• NVlabs / ConvSSM
  ☆69Updated last year
• UCSC-VLAA / CRATE-alpha
  This repository includes the official implementation our paper "Scaling White-Box Transformers for Vision"
  ☆48Updated last year
• bwconrad / soft-moe
  PyTorch implementation of "From Sparse to Soft Mixtures of Experts"
  ☆68Updated 2 years ago
• kaiyuyue / nxtp
  PyTorch Implementation of Object Recognition as Next Token Prediction [CVPR'24 Highlight]
  ☆182Updated 9 months ago
• bfshi / TOAST
  Official code for "TOAST: Transfer Learning via Attention Steering"
  ☆188Updated 2 years ago
• Hprairie / Bi-Mamba2
  A Triton Kernel for incorporating Bi-Directionality in Mamba2
  ☆78Updated last year
• CompVis / discrete-interpolants
  The official implementation of "[MASK] is All You Need"
  ☆126Updated 6 months ago