/mamba.py

A simple and efficient Mamba implementation in pure PyTorch and MLX.

Primary LanguagePythonMIT LicenseMIT

Stargazers