roikoren755 b897f00c9c Gate SSU dispatch setup (#40039) ...

Signed-off-by: Roi Koren <roik@nvidia.com>

2026-04-16 13:06:01 -07:00

..

ops

Gate SSU dispatch setup (#40039)

2026-04-16 13:06:01 -07: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

[Model][Spec Decode] Nemotron-H MTP and Mamba Speculative Decoding Support (#33726)

2026-02-24 09:49:56 -08:00

gdn_linear_attn.py

[compile] Allow strings in custom ops without regressing compilation times (#38123)

2026-04-10 07:26:37 +00:00

lamport_workspace.py

[Kernel] Porting the TRTLLM minimax_allreduce_rms kernels (#37045)

2026-04-11 00:20:20 +08:00

linear_attn.py

[V0 Deprecation] Refactor kv cache from list to element (#37487)

2026-03-23 20:10:11 -07:00

mamba_mixer2.py

[Mamba] Flashinfer selective_state_update (#36162)

2026-04-14 15:10:58 -04:00

mamba_mixer.py

[Mamba] Flashinfer selective_state_update (#36162)

2026-04-14 15:10:58 -04:00

mamba_utils.py

[Kernel] Mamba support different layout for Conv state (#37416)

2026-04-03 01:50:09 +02:00

short_conv.py

[Kernel] Mamba support different layout for Conv state (#37416)

2026-04-03 01:50:09 +02:00