Alternatives and similar repositories for s4:

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

• alxndrTL / mamba.py
  A simple and efficient Mamba implementation in pure PyTorch and MLX.
  ☆1,125Updated 2 months ago
• lucidrains / rotary-embedding-torch
  Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch
  ☆635Updated 2 months ago
• srush / annotated-s4
  Implementation of https://srush.github.io/annotated-s4
  ☆483Updated 2 years ago
• johnma2006 / mamba-minimal
  Simple, minimal implementation of the Mamba SSM in one file of PyTorch.
  ☆2,717Updated 11 months ago
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆738Updated 9 months ago
• lucidrains / vector-quantize-pytorch
  Vector (and Scalar) Quantization, in Pytorch
  ☆2,922Updated last week
• lucidrains / lion-pytorch
  🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch
  ☆2,094Updated 2 months ago
• idiap / fast-transformers
  Pytorch library for fast transformer implementations
  ☆1,677Updated last year
• state-spaces / mamba
  Mamba SSM architecture
  ☆14,009Updated last month
• Jamie-Stirling / RetNet
  An implementation of "Retentive Network: A Successor to Transformer for Large Language Models"
  ☆1,173Updated last year
• lucidrains / x-transformers
  A concise but complete full-attention transformer with a set of promising experimental features from various papers
  ☆5,080Updated this week
• facebookresearch / schedule_free
  Schedule-Free Optimization in PyTorch
  ☆2,098Updated 2 months ago
• lucidrains / perceiver-pytorch
  Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch
  ☆1,120Updated last year
• yyyujintang / Awesome-Mamba-Papers
  Awesome Papers related to Mamba.
  ☆1,306Updated 4 months ago
• radarFudan / Awesome-state-space-models
  Collection of papers on state-space models
  ☆575Updated 3 weeks ago
• test-time-training / ttt-lm-pytorch
  Official PyTorch implementation of Learning to (Learn at Test Time): RNNs with Expressive Hidden States
  ☆1,126Updated 7 months ago
• google-research / long-range-arena
  Long Range Arena for Benchmarking Efficient Transformers
  ☆745Updated last year
• fla-org / flash-linear-attention
  🚀 Efficient implementations of state-of-the-art linear attention models in Torch and Triton
  ☆1,912Updated this week
• microsoft / torchscale
  Foundation Architecture for (M)LLMs
  ☆3,046Updated 10 months ago
• NX-AI / xlstm
  Official repository of the xLSTM.
  ☆1,692Updated last month
• microsoft / mup
  maximal update parametrization (µP)
  ☆1,451Updated 7 months ago
• lindermanlab / S5
  ☆280Updated last month
• lucidrains / performer-pytorch
  An implementation of Performer, a linear attention-based transformer, in Pytorch
  ☆1,115Updated 3 years ago
• google / learned_optimization
  ☆764Updated last month
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆1,802Updated this week
• n2cholas / awesome-jax
  JAX - A curated list of resources https://github.com/google/jax
  ☆1,699Updated this week
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆473Updated 11 months ago
• clementchadebec / benchmark_VAE
  Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)
  ☆1,872Updated 6 months ago
• yang-song / score_sde
  Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
  ☆1,571Updated 2 years ago
• webdataset / webdataset
  A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.
  ☆2,463Updated last week