Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)
Primary LanguagePythonMozilla Public License 2.0MPL-2.0