roikoren755andzjy0516 e11962fc1c [Feat][Mamba2] Enable internal prefill checkpoints (#57329) ...

Signed-off-by: Roi Koren <roik@nvidia.com>
Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
Co-authored-by: zjy0516 <riverclouds.zhu@qq.com>

2026-10-06 07:53:51 +00:00

..

gdn

[Perf] Keep non-speculative GDN decode on the standard path (#59735)

2026-10-02 13:44:53 +00:00

linear

[Chore] Use Transformers v5 names and drop redundant processor use_fast (#58550)

2026-09-24 14:14:10 +00:00

ops

[Feat][Mamba2] Enable internal prefill checkpoints (#57329)

2026-10-06 07:53:51 +00:00

__init__.py

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

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

abstract.py

[Mamba] Add FlashInfer ReplaySSM support for MTP (#52928)

2026-09-30 21:41:17 +00:00

checkpoint.py

[Feat][Mamba2] Enable internal prefill checkpoints (#57329)

2026-10-06 07:53:51 +00:00

kda_checkpoint.py

[Core][KDA] Generalize Mamba prefill checkpoint builder and exporter (#57783)

2026-09-21 04:46:11 +00:00

mamba_mixer2.py

[Feat][Mamba2] Enable internal prefill checkpoints (#57329)

2026-10-06 07:53:51 +00:00

mamba_mixer.py

[Core] Remove deprecated mamba_cache_mode "all" (#58997)

2026-10-01 02:59:37 -07:00

mamba_utils.py

[Mamba] Add FlashInfer ReplaySSM support for MTP (#52928)

2026-09-30 21:41:17 +00:00

short_conv.py

[Kernel] Remove unused fake implementation (#55535)

2026-09-06 17:54:32 -07:00