Simple, minimal implementation of the Mamba SSM in one file of PyTorch. - View it on GitHub