johnma2006/mamba-minimal

Simple, minimal implementation of the Mamba SSM in one file of PyTorch.

语言：Python · GitHub 星标：2,969 · Fork：219

在 GitHub 上查看原仓库