mamba-from-scratch
From-scratch implementation of S4, Mamba-1 (S6), and Mamba-2 (SSD) state-space models in pure PyTorch. Clear, well-tested, educational.
原文来源AI 开源 Releases
查看官方原文
From-scratch implementation of S4, Mamba-1 (S6), and Mamba-2 (SSD) state-space models in pure PyTorch. Clear, well-tested, educational.