☆317Jan 8, 2025Updated last year
Alternatives and similar repositories for S5
Users that are interested in S5 are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)☆83Apr 26, 2024Updated last year
- Structured state space sequence models☆2,875Jul 17, 2024Updated last year
- Implementation of https://srush.github.io/annotated-s4☆515Jun 20, 2025Updated 9 months ago
- ☆165Jan 24, 2023Updated 3 years ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆62Sep 3, 2025Updated 7 months ago
- NordVPN Special Discount Offer • AdSave on top-rated NordVPN 1 or 2-year plans with secure browsing, privacy protection, and support for for all major platforms.
- 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☆391Feb 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 11 months ago
- A State-Space Model with Rational Transfer Function Representation.☆83May 17, 2024Updated last year
- Convolutions for Sequence Modeling☆911Jun 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
- NordVPN Special Discount Offer • AdSave on top-rated NordVPN 1 or 2-year plans with secure browsing, privacy protection, and support for for all major platforms.
- Scalable and Stable Parallelization of Nonlinear RNNS☆30Mar 6, 2026Updated last month
- ☆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☆13Apr 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
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆21Jul 14, 2024Updated last year
- Experiments on the impact of depth in transformers and SSMs.☆41Oct 23, 2025Updated 5 months ago
- NordVPN Special Discount Offer • AdSave on top-rated NordVPN 1 or 2-year plans with secure browsing, privacy protection, and support for for all major platforms.
- Parallelizing non-linear sequential models over the sequence length☆57Jun 23, 2025Updated 9 months ago
- HGRN2: Gated Linear RNNs with State Expansion☆57Aug 20, 2024Updated last year
- Accelerated First Order Parallel Associative Scan☆197Jan 7, 2026Updated 3 months ago
- Implementations of various linear RNN layers using pytorch and triton☆55Aug 4, 2023Updated 2 years ago
- ☆11Oct 11, 2023Updated 2 years ago
- Collection of papers on state-space models☆618Nov 4, 2025Updated 5 months ago
- Annotated version of the Mamba paper☆500Feb 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
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆16Jan 7, 2025Updated last year
- ☆45Apr 30, 2018Updated 7 years ago
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆102Feb 25, 2023Updated 3 years ago
- A Python package for probabilistic state space modeling with JAX☆949Jan 6, 2026Updated 3 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
- ☆53Jan 19, 2023Updated 3 years ago
- ☆199May 27, 2024Updated last year