search

johnma2006 / mamba-minimal

Simple, minimal implementation of the Mamba SSM in one file of PyTorch.
Apache License 2.0
2.54k stars 188 forks source link

Mamba-2 implementation #31

Open tastelessjolt opened 1 month ago

tastelessjolt commented 1 month ago

Hello, is there any plan to make a simple educational version of the Mamba-2 architecture implementation?

tastelessjolt commented 1 month ago

Found one pure pytorch implementation of Mamba2 architecture: https://github.com/vasqu/mamba2-torch/