The accompanying code for "Simplifying and Understanding State Space Models with Diagonal Linear RNNs" (Ankit Gupta, Harsh Mehta, Jonathan Berant).
Primary LanguageJupyter NotebookApache License 2.0Apache-2.0