Alternatives and similar repositories for SSM-Bottleneck

Users that are interested in SSM-Bottleneck are comparing it to the libraries listed below

• HelmholtzAI-FZJ / flex_gen
  ☆17Updated 4 months ago
• OpenNLPLab / HGRN2
  HGRN2: Gated Linear RNNs with State Expansion
  ☆54Updated 9 months ago
• smonsays / hypernetwork-attention
  Official code for the paper "Attention as a Hypernetwork"
  ☆36Updated 11 months ago
• kyegomez / Hedgehog
  Implementation of the model "Hedgehog" from the paper: "The Hedgehog & the Porcupine: Expressive Linear Attentions with Softmax Mimicry"
  ☆13Updated last year
• RobertCsordas / linear_layer_as_attention
  The official repository for our paper "The Dual Form of Neural Networks Revisited: Connecting Test Time Predictions to Training Patterns …
  ☆16Updated last year
• ermongroup / fast_feedforward_computation
  Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021
  ☆27Updated 3 years ago
• glassroom / heinsen_attention
  Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)
  ☆24Updated 11 months ago
• radarFudan / Curse-of-memory
  Curse-of-memory phenomenon of RNNs in sequence modelling
  ☆19Updated 3 weeks ago
• Eliyas0007 / Pytorch-Intention
  Unofficial implementation of paper : Exploring the Space of Key-Value-Query Models with Intention
  ☆11Updated 2 years ago
• bdusell / stack-attention
  Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"
  ☆17Updated last year
• MarkXCloud / CSpD
  The official repo of continuous speculative decoding
  ☆26Updated 2 months ago
• philippe-eecs / small-vision
  A big_vision inspired repo that implements a generic Auto-Encoder class capable in representation learning and generative modeling.
  ☆35Updated 11 months ago
• hengyuan-hu / jax-vs-pytorch
  ☆12Updated 3 months ago
• kyegomez / MAGVIT2
  Open source community's implementation of the model from "LANGUAGE MODEL BEATS DIFFUSION — TOKENIZER IS KEY TO VISUAL GENERATION"
  ☆16Updated 6 months ago
• Doraemonzzz / hgru2-pytorch
  ☆23Updated 8 months ago
• dangxingyu / rnn-icrag
  Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"
  ☆27Updated last year
• TIGER-AI-Lab / VISTA
  The code for "VISTA: Enhancing Long-Duration and High-Resolution Video Understanding by VIdeo SpatioTemporal Augmentation" [CVPR2025]
  ☆15Updated 3 months ago
• facebookresearch / dmae_st
  Directed masked autoencoders
  ☆14Updated 2 years ago
• showlab / D-AR
  the official repo for "D-AR: Diffusion via Autoregressive Models"
  ☆30Updated last week
• lzzcd001 / nabla-gfn
  Official Implementation of Nabla-GFlowNet (ICLR 2025)
  ☆19Updated last month
• NVlabs / TCM
  Codebase of Truncated Consistency Models (ICLR 2025)
  ☆23Updated 4 months ago
• yuweihao / LV-BERT
  LV-BERT: Exploiting Layer Variety for BERT (Findings of ACL 2021)
  ☆18Updated 2 years ago
• apple / ml-agm
  ☆44Updated last year
• ziplab / SN-Netv2
  [ECCV 2024] This is the official implementation of "Stitched ViTs are Flexible Vision Backbones".
  ☆27Updated last year
• wjxts / RegularizedBN
  ☆21Updated 2 years ago
• CompVis / DisCLIP
  Does VLM Classification Benefit from LLM Description Semantics? (AAAI 2025)
  ☆17Updated 4 months ago
• zhixuan-lin / forgetting-transformer
  [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"
  ☆104Updated 3 weeks ago
• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆23Updated last year
• Optimization-AI / FastCLIP
  Distributed Optimization Infra for learning CLIP models
  ☆26Updated 8 months ago
• akhilkedia / TranformersGetStable
  [ICML 2024] Official Repository for the paper "Transformers Get Stable: An End-to-End Signal Propagation Theory for Language Models"
  ☆10Updated 10 months ago