├── README.md └── models ├── mamba.py ├── mamba2.py ├── mamba2_equinox.py ├── ssm_attention_duality.txt └── xlstm.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okarthikb/state-space-models/HEAD/README.md -------------------------------------------------------------------------------- /models/mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okarthikb/state-space-models/HEAD/models/mamba.py -------------------------------------------------------------------------------- /models/mamba2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okarthikb/state-space-models/HEAD/models/mamba2.py -------------------------------------------------------------------------------- /models/mamba2_equinox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okarthikb/state-space-models/HEAD/models/mamba2_equinox.py -------------------------------------------------------------------------------- /models/ssm_attention_duality.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okarthikb/state-space-models/HEAD/models/ssm_attention_duality.txt -------------------------------------------------------------------------------- /models/xlstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okarthikb/state-space-models/HEAD/models/xlstm.py --------------------------------------------------------------------------------