Alternatives and similar repositories for s4

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

• johnma2006 / mamba-minimal
  Simple, minimal implementation of the Mamba SSM in one file of PyTorch.
  ☆2,822Updated last year
• idiap / fast-transformers
  Pytorch library for fast transformer implementations
  ☆1,720Updated 2 years ago
• srush / annotated-s4
  Implementation of https://srush.github.io/annotated-s4
  ☆499Updated last week
• alxndrTL / mamba.py
  A simple and efficient Mamba implementation in pure PyTorch and MLX.
  ☆1,265Updated 6 months ago
• lucidrains / lion-pytorch
  🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch
  ☆2,143Updated 7 months ago
• microsoft / mup
  maximal update parametrization (µP)
  ☆1,549Updated 11 months ago
• lucidrains / rotary-embedding-torch
  Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch
  ☆699Updated 7 months ago
• lucidrains / x-transformers
  A concise but complete full-attention transformer with a set of promising experimental features from various papers
  ☆5,413Updated this week
• radarFudan / Awesome-state-space-models
  Collection of papers on state-space models
  ☆595Updated 2 months ago
• google / learned_optimization
  ☆781Updated 3 weeks ago
• google-research / long-range-arena
  Long Range Arena for Benchmarking Efficient Transformers
  ☆758Updated last year
• facebookresearch / schedule_free
  Schedule-Free Optimization in PyTorch
  ☆2,185Updated last month
• clementchadebec / benchmark_VAE
  Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)
  ☆1,920Updated 11 months ago
• lucidrains / vector-quantize-pytorch
  Vector (and Scalar) Quantization, in Pytorch
  ☆3,349Updated 2 weeks ago
• microsoft / torchscale
  Foundation Architecture for (M)LLMs
  ☆3,083Updated last year
• webdataset / webdataset
  A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.
  ☆2,658Updated last week
• state-spaces / mamba
  Mamba SSM architecture
  ☆15,204Updated this week
• facebookresearch / flow_matching
  A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…
  ☆2,897Updated last month
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆781Updated last year
• lucidrains / mixture-of-experts
  A Pytorch implementation of Sparsely-Gated Mixture of Experts, for massively increasing the parameter count of language models
  ☆770Updated last year
• lucidrains / reformer-pytorch
  Reformer, the efficient Transformer, in Pytorch
  ☆2,171Updated 2 years ago
• fla-org / flash-linear-attention
  🚀 Efficient implementations of state-of-the-art linear attention models in Torch and Triton
  ☆2,811Updated this week
• lucidrains / performer-pytorch
  An implementation of Performer, a linear attention-based transformer, in Pytorch
  ☆1,134Updated 3 years ago
• mert-kurttutan / torchview
  torchview: visualize pytorch models
  ☆957Updated last month
• AvivBick / awesome-ssm-ml
  Reading list for research topics in state-space models
  ☆298Updated 2 weeks ago
• ml-jku / hopfield-layers
  Hopfield Networks is All You Need
  ☆1,814Updated 2 years ago
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆9,001Updated this week
• google / flax
  Flax is a neural network library for JAX that is designed for flexibility.
  ☆6,647Updated this week
• ashleve / lightning-hydra-template
  PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡🔥⚡
  ☆4,725Updated 10 months ago
• google-research / big_vision
  Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.
  ☆2,964Updated last month