state-spaces/s4
A PyTorch library implementing structured state space sequence models (S4, HiPPO, LSSL) for efficient sequence modeling.
Velocity · 7d
+1.7
★ / day
Trend
→steady
star history
This repository provides official implementations of structured state space models for sequence modeling, including S4, HiPPO, LSSL, SaShiMi, DSS, and related variants. The models leverage efficient kernel computations (Cauchy and Vandermonde) implemented in PyTorch for various sequence tasks including generation and training from scratch.