A simple and efficient Mamba implementation in pure PyTorch and MLX. - View it on GitHub