TensorRT-LLMs/tensorrt_llm/_torch/modules/mamba
Iman Tabrizian 7d992972b2
[TRTLLM-10273][feat] Move MambaCacheManager from Python to C++ (#10540)
Signed-off-by: Iman Tabrizian <10105175+tabrizian@users.noreply.github.com>
2026-02-10 07:20:56 -08:00
..
__init__.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00
causal_conv1d_triton.py [TRTLLM-10062][feat] Enable MTP for Nemotron Super (#10754) 2026-01-26 11:23:26 -05:00
causal_conv1d.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00
layernorm_gated.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00
mamba2_metadata.py [TRTLLM-10273][feat] Move MambaCacheManager from Python to C++ (#10540) 2026-02-10 07:20:56 -08:00
mamba2_mixer.py [TRTLLM-10273][feat] Move MambaCacheManager from Python to C++ (#10540) 2026-02-10 07:20:56 -08:00
selective_state_update.py [None][fix] Fix selective_state_update perf regression for T=1 decode path (#11194) 2026-02-04 09:01:34 +02:00
softplus.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00
ssd_bmm.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00
ssd_chunk_scan.py [TRTLLM-10062][feat] Enable MTP for Nemotron Super (#10754) 2026-01-26 11:23:26 -05:00
ssd_chunk_state.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00
ssd_combined.py [TRTLLM-10062][feat] Enable MTP for Nemotron Super (#10754) 2026-01-26 11:23:26 -05:00
ssd_state_passing.py [TRTLLM-4921][feat] Enable chunked prefill for Nemotron-H (#6334) 2025-08-22 12:15:20 -04:00