elephantmipt/annotated-mamba

Annotated version of the Mamba paper

Part 1:

Blog: https://srush.github.io/annotated-mamba/hard.html

Notebook: https://github.com/srush/annotated-mamba/blob/main/Scan.ipynb

Mamba: Linear-Time Sequence Modeling with Selective State Spaces https://arxiv.org/abs/2312.00752