☆316Jan 8, 2025Updated last year
Alternatives and similar repositories for S5
Users that are interested in S5 are comparing it to the libraries listed below
Sorting:
- Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)☆82Apr 26, 2024Updated last year
- Structured state space sequence models☆2,869Jul 17, 2024Updated last year
- Implementation of https://srush.github.io/annotated-s4☆513Jun 20, 2025Updated 9 months ago
- ☆164Jan 24, 2023Updated 3 years ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆62Sep 3, 2025Updated 6 months ago
- Sequence Modeling with Structured State Spaces☆67Aug 2, 2022Updated 3 years ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆112Dec 5, 2023Updated 2 years ago
- The accompanying code for "Simplifying and Understanding State Space Models with Diagonal Linear RNNs" (Ankit Gupta, Harsh Mehta, Jonatha…☆23Dec 30, 2022Updated 3 years ago
- Liquid Structural State-Space Models☆389Feb 1, 2024Updated 2 years ago
- Sequence modeling with Mega.☆303Jan 28, 2023Updated 3 years ago
- Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch☆74Apr 22, 2025Updated 10 months ago
- A State-Space Model with Rational Transfer Function Representation.☆83May 17, 2024Updated last year
- Convolutions for Sequence Modeling☆912Jun 13, 2024Updated last year
- Code for SpaceTime 🌌⏱️. Proposed in Effectively Modeling Time Series with Simple Discrete State Spaces, ICLR 2023.☆181Mar 17, 2023Updated 3 years ago
- Scalable and Stable Parallelization of Nonlinear RNNS☆29Mar 6, 2026Updated 2 weeks ago
- ☆35Nov 22, 2024Updated last year
- Implementation of the "Online learning of long-range dependencies" paper, NeurIPS 2023☆21Nov 4, 2024Updated last year
- POPGym Library in JAX☆12Apr 15, 2024Updated last year
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆68Apr 24, 2024Updated last year
- ☆35Apr 12, 2024Updated last year
- ☆20Apr 17, 2023Updated 2 years ago
- Experiments on the impact of depth in transformers and SSMs.☆41Oct 23, 2025Updated 4 months ago
- Parallelizing non-linear sequential models over the sequence length☆55Jun 23, 2025Updated 8 months ago
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆21Jul 14, 2024Updated last year
- HGRN2: Gated Linear RNNs with State Expansion☆56Aug 20, 2024Updated last year
- Accelerated First Order Parallel Associative Scan☆196Jan 7, 2026Updated 2 months ago
- Implementations of various linear RNN layers using pytorch and triton☆54Aug 4, 2023Updated 2 years ago
- ☆11Oct 11, 2023Updated 2 years ago
- Collection of papers on state-space models☆616Nov 4, 2025Updated 4 months ago
- Annotated version of the Mamba paper☆499Feb 27, 2024Updated 2 years ago
- ☆29Jul 9, 2024Updated last year
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆98Dec 5, 2024Updated last year
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Jan 7, 2025Updated last year
- ☆45Apr 30, 2018Updated 7 years ago
- A Python package for probabilistic state space modeling with JAX☆940Jan 6, 2026Updated 2 months ago
- PyTorch implementation of the NCDSSM models presented in the ICML '23 paper "Neural Continuous-Discrete State Space Models for Irregularl…☆26Jul 9, 2023Updated 2 years ago
- ☆52Jan 19, 2023Updated 3 years ago
- ☆198May 27, 2024Updated last year
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆57Dec 4, 2024Updated last year