mamba_ssm/__init__.py

__version__ = "2.0.3"

from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
from .modules.mamba_simple import Mamba
from .modules.mamba2 import Mamba2
from .models.mixer_seq_simple import MambaLMHeadModel