A simple and efficient Mamba implementation in pure PyTorch and MLX.
Primary LanguagePythonMIT LicenseMIT