.gitignore
.pre-commit-config.yaml
AUTHORS
CITATION.cff
LICENSE
MANIFEST.in
Makefile
README.md
pyproject.toml
requirements.txt
uv.lock
.github/workflows/lint.yml
.github/workflows/publish.yml
.github/workflows/test.yml
examples/01_core_modules.py
examples/02_text_generation.py
examples/03_training.py
examples/03_training.yaml
examples/__init__.py
mamba_ssm/__init__.py
mamba_ssm/models/__init__.py
mamba_ssm/models/config_mamba.py
mamba_ssm/models/mixer_seq_simple.py
mamba_ssm/modules/__init__.py
mamba_ssm/modules/block.py
mamba_ssm/modules/mamba2.py
mamba_ssm/modules/mamba_simple.py
mamba_ssm/modules/mha.py
mamba_ssm/modules/mlp.py
mamba_ssm/ops/__init__.py
mamba_ssm/ops/selective_scan_interface.py
mamba_ssm/utils/__init__.py
mamba_ssm/utils/generation.py
mamba_ssm/utils/hf.py
mamba_ssm/utils/macos.py
mamba_ssm/utils/torch.py
mamba_ssm_macos.egg-info/PKG-INFO
mamba_ssm_macos.egg-info/SOURCES.txt
mamba_ssm_macos.egg-info/dependency_links.txt
mamba_ssm_macos.egg-info/requires.txt
mamba_ssm_macos.egg-info/top_level.txt
scripts/download_models.py
scripts/run_models.py
tests/__init__.py
tests/run_unit_tests.py
tests/integration/__init__.py
tests/integration/test_unified_system.py
tests/unit/__init__.py
tests/unit/test_generation_macos.py
tests/unit/test_mamba2_macos.py
tests/unit/test_mamba_macos.py
tests/unit/test_selective_scan_gradients.py