Andrii SkliarandAndrii Skliar deb148e730 [Mamba] Add FlashInfer ReplaySSM support for MTP (#52928) ...

Signed-off-by: Andrii Skliar <askliar@nvidia.com>
Co-authored-by: Andrii Skliar <askliar@nvidia.com>

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

..

attention

[KV-Offloading][TP] : Expand replicated_layout detection to multi-group MLA (#57652)

2026-09-30 09:11:03 -04:00

fused_moe

[ROCm]Transpose compressed-tensors MoE weights on device (#59253)

2026-09-30 12:44:40 -05:00

fusion

[MM] Keep device input normalization fused when encoder compilation is enabled (#59195)

2026-09-30 11:08:24 +00:00

hpc

Add pydocstyle to the ruff rules (#52136)

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

mamba

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

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

minimax_rms_norm

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

pooler

[LoRA] Support variable num_labels for sequence classification (#57766)

2026-09-28 14:58:35 +00:00

quantization

[Quantization] Add per-token NVFP4 CuTe-DSL MoE backend (#50030)

2026-09-30 10:42:03 +00:00

rotary_embedding

[CI][Bugfix] Relax packed_qk_rope_ correctness test to one ULP (#59008)

2026-09-28 17:29:59 +00:00

__init__.py

…

activation.py

[ROCm] Use silu_and_mul_with_clamp's torch._C op (#52052)

2026-09-22 18:30:14 -05:00

attention_layer_base.py

Add pydocstyle to the ruff rules (#52136)

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

conv.py

Add pydocstyle to the ruff rules (#52136)

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

fused_allreduce_gemma_rms_norm.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

fused_embed_norm.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

fused_qk_norm_rope.py

Add pydocstyle to the ruff rules (#52136)

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

indexer_topk.py

[GLM-5.3-Flash] Route kpool indexer top-k through the shared SparseIndexerTopk dispatcher (#57546)

2026-09-20 11:17:59 +00:00

layernorm.py

[XPU] Dispatch nn.LayerNorm to fused SYCL kernel via CustomOp (#57172)

2026-09-30 12:19:33 +00:00

lightning_attn.py

Add pydocstyle to the ruff rules (#52136)

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

linear.py

[ROCm][Bugfix] Revert #56433 + #51692 to fix accuracy breakdown for DeepSeek-V4 (#57132)

2026-09-16 18:58:23 +00:00

logits_processor.py

[SpecDecode] Add LiLiCorr drafter (#57934)

2026-09-25 10:08:28 -04:00

mhc.py

[ROCm][DSv4.1][Perf] Run the delayed mHC seams through aiter's fused Triton kernel (#58655)

2026-09-29 10:19:10 +08:00

mla.py

[3/N] HiSparse: host-resident sparse-MLA decode hot-buffering (#53781)

2026-09-12 00:36:47 -04:00

resampler.py

Add pydocstyle to the ruff rules (#52136)

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

rocm_paged_mxfp4_indexer.py

[ROCm][DSv4.1] Paged MXFP4 sparse indexer on aiter's MQA-logits kernel (#58671)

2026-09-29 14:37:57 +08:00

sparse_attn_indexer.py

[Bugfix] Take padded path for ragged decode batches in sparse_attn_indexer (#52500)

2026-09-19 20:14:29 +00:00

sparse_mqa_indexer.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

utils.py

[CPU] Conv1d optimised kernel for aarch64 (#54093)

2026-09-30 20:23:42 +08:00

vocab_parallel_embedding.py

[Bugfix][Quantization] Give LM heads standard linear metadata (#58444)

2026-09-24 16:39:49 -07:00