Yu Chin Fabian Lim fca20841c2 Correction to TP logic for Mamba Mixer 2 when Num Groups not divisible by TP Size (#13660)

2025-02-22 00:19:10 -08:00

..

ops

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

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

mamba_mixer2.py

Correction to TP logic for Mamba Mixer 2 when Num Groups not divisible by TP Size (#13660)

2025-02-22 00:19:10 -08:00

mamba_mixer.py

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

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