Samuel KimandClaude 34ee85d5ca [Perf] Batch Mamba2 prefill SSM state saves, removing GPU<->CPU syncs (#49371) ...

Signed-off-by: samuelkim7 <samuelmwkim@gmail.com>
Co-authored-by: Claude <noreply@anthropic.com>

2026-09-25 01:11:19 +00:00

..

gdn

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

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

linear

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

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

ops

[CPU][GDN] Support NIXL DS convolution-state layout (#53300)

2026-09-24 14:55:20 +08: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

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

checkpoint.py

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

2026-09-21 04:46:11 +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

[Perf] Batch Mamba2 prefill SSM state saves, removing GPU<->CPU syncs (#49371)

2026-09-25 01:11:19 +00:00

mamba_mixer.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

mamba_utils.py

[Mypy] Fix mypy typing for N/O models (#54142)

2026-09-18 12:50:01 +01:00

short_conv.py

[Kernel] Remove unused fake implementation (#55535)

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