Chih-Chieh YangandYu Chin Fabian Lim daefed052c [Model] Reduce redundant computations in mamba2 blocks for Bamba-9B (#15423) ...

Signed-off-by: Chih-Chieh-Yang <7364402+cyang49@users.noreply.github.com>
Co-authored-by: Yu Chin Fabian Lim <flim@sg.ibm.com>

2025-04-10 19:07:07 +00:00

..

ops

[Model] Reduce redundant computations in mamba2 blocks for Bamba-9B (#15423)

2025-04-10 19:07:07 +00:00

__init__.py

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

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

mamba2_metadata.py

[Model] Reduce redundant computations in mamba2 blocks for Bamba-9B (#15423)

2025-04-10 19:07:07 +00:00

mamba_mixer2.py

[Model] Reduce redundant computations in mamba2 blocks for Bamba-9B (#15423)

2025-04-10 19:07:07 +00:00

mamba_mixer.py

Remove unused kwargs from model definitions (#13555)

2025-02-24 17:13:52 -08:00