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,384Updated last year
• idiap / fast-transformers
  Pytorch library for fast transformer implementations
  ☆1,755Updated 2 years ago
• johnma2006 / mamba-minimal
  Simple, minimal implementation of the Mamba SSM in one file of PyTorch.
  ☆2,901Updated last year
• srush / annotated-s4
  Implementation of https://srush.github.io/annotated-s4
  ☆507Updated 5 months ago
• lucidrains / rotary-embedding-torch
  Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch
  ☆782Updated 4 months ago
• Jamie-Stirling / RetNet
  An implementation of "Retentive Network: A Successor to Transformer for Large Language Models"
  ☆1,210Updated 2 years ago
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆815Updated last year
• lucidrains / performer-pytorch
  An implementation of Performer, a linear attention-based transformer, in Pytorch
  ☆1,168Updated 3 years ago
• google-research / long-range-arena
  Long Range Arena for Benchmarking Efficient Transformers
  ☆769Updated last year
• lucidrains / lion-pytorch
  🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch
  ☆2,177Updated last year
• microsoft / mup
  maximal update parametrization (µP)
  ☆1,638Updated last year
• lucidrains / perceiver-pytorch
  Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch
  ☆1,183Updated 2 years ago
• lucidrains / reformer-pytorch
  Reformer, the efficient Transformer, in Pytorch
  ☆2,191Updated 2 years ago
• lucidrains / x-transformers
  A concise but complete full-attention transformer with a set of promising experimental features from various papers
  ☆5,710Updated last month
• ml-jku / hopfield-layers
  Hopfield Networks is All You Need
  ☆1,882Updated 2 years ago
• microsoft / torchscale
  Foundation Architecture for (M)LLMs
  ☆3,125Updated last year
• NX-AI / xlstm
  Official repository of the xLSTM.
  ☆2,059Updated last month
• lucidrains / vector-quantize-pytorch
  Vector (and Scalar) Quantization, in Pytorch
  ☆3,756Updated this week
• facebookresearch / schedule_free
  Schedule-Free Optimization in PyTorch
  ☆2,237Updated 6 months ago
• HazyResearch / safari
  Convolutions for Sequence Modeling
  ☆908Updated last year
• davidmrau / mixture-of-experts
  PyTorch Re-Implementation of "The Sparsely-Gated Mixture-of-Experts Layer" by Noam Shazeer et al. https://arxiv.org/abs/1701.06538
  ☆1,210Updated last year
• fla-org / flash-linear-attention
  🚀 Efficient implementations of state-of-the-art linear attention models
  ☆4,049Updated this week
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆1,053Updated 5 months ago
• lindermanlab / S5
  ☆312Updated 11 months ago
• google / learned_optimization
  ☆789Updated 3 weeks ago
• facebookresearch / multimodal
  TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-task models at scale.
  ☆1,675Updated 2 weeks ago
• lucidrains / local-attention
  An implementation of local windowed attention for language modeling
  ☆489Updated 4 months ago
• state-spaces / mamba
  Mamba SSM architecture
  ☆16,646Updated last month
• Liuhong99 / Sophia
  The official implementation of “Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training”
  ☆980Updated last year
• atong01 / conditional-flow-matching
  TorchCFM: a Conditional Flow Matching library
  ☆2,174Updated last month