Tyler Michael SmithandYu Chin Fabian Lim 1f69c4a892 [Model] Support Mamba2 (Codestral Mamba) (#9292) ...

Signed-off-by: Tyler Michael Smith <tyler@neuralmagic.com>
Co-authored-by: Yu Chin Fabian Lim <flim@sg.ibm.com>

2025-02-17 20:17:50 +08:00

..

__init__.py

[Kernel/Model] Migrate mamba_ssm and causal_conv1d kernels to vLLM (#7651)

2024-08-28 15:06:52 -07:00

causal_conv1d.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

mamba_ssm.py

Add Bamba Model (#10909)

2025-02-06 15:22:42 -08:00

ssd_bmm.py

Add Bamba Model (#10909)

2025-02-06 15:22:42 -08:00

ssd_chunk_scan.py

[Model] Support Mamba2 (Codestral Mamba) (#9292)

2025-02-17 20:17:50 +08:00

ssd_chunk_state.py

Add Bamba Model (#10909)

2025-02-06 15:22:42 -08:00

ssd_combined.py

Add Bamba Model (#10909)

2025-02-06 15:22:42 -08:00

ssd_state_passing.py

Add Bamba Model (#10909)

2025-02-06 15:22:42 -08:00