3 people a4d70bef37 [Model Runner V2] Reserve CUDA graph memory (#53306) ...

Signed-off-by: Anh Tran <anh.tran.3889@gmail.com>
Signed-off-by: Nick Hill <nickhill123@gmail.com>
Co-authored-by: Anh Tran <anh.tran.3889@gmail.com>
Co-authored-by: anhtra3889 <anhtra@nvidia.com>

2026-08-23 23:11:19 -07:00

..

attention

[6/N][KV-Cache Layout Refactor] Standardize KV cache layout (#51718)

2026-08-21 17:47:49 -07:00

fused_moe

[LoRA] feat: Support LoRA for DeepSeek V4 (#53361)

2026-08-24 03:57:57 +00:00

fusion

Fix docs build on main (#45536)

2026-06-13 08:53:23 -07:00

hpc

[6/N][KV-Cache Layout Refactor] Standardize KV cache layout (#51718)

2026-08-21 17:47:49 -07:00

mamba

[Model Runner V2] Reserve CUDA graph memory (#53306)

2026-08-23 23:11:19 -07:00

minimax_rms_norm

[ROCm][MiniMax-M2] Dispatch fused QK-norm + AllReduce via AITER (#44849)

2026-07-14 03:10:16 +00:00

pooler

[ModelRunner v2] Enable decoder token-wise pooling (#50931)

2026-08-06 22:21:52 +00:00

quantization

[Bugfix][ROCM] Fix the MXFP8 block scale exponent (#53110)

2026-08-22 01:26:19 +00:00

rotary_embedding

[XPU] follow cuda path for mrope on XPU (#53201)

2026-08-21 10:13:27 +00:00

__init__.py

…

activation.py

[Refactor] Remove multiple dead codes (#50285)

2026-08-03 10:59:08 -04:00

attention_layer_base.py

[3/N][KV-Cache Layout Refactor] Standardize Mamba cache; drop get_transfer_cache_regions (#44456)

2026-07-21 09:16:15 +00:00

batch_invariant.py

[Bugfix] Fix batch-invariant fp32 matmul OOR on SM89 for N=1 (#52960)

2026-08-21 11:46:55 +08:00

conv.py

[Refactor] Remove multiple dead codes (#50285)

2026-08-03 10:59:08 -04:00

fused_allreduce_gemma_rms_norm.py

[Bugfix] Re-land MiniMax M3 default video processor (#50305)

2026-07-31 07:46:18 +00:00

fused_embed_norm.py

Replicated embedding and norm fusion for DSV3 flat model (#48484)

2026-08-18 14:53:21 +08:00

fused_qk_norm_rope.py

[Perf] fuse qk rmsnorm rope gate for qwen3.5 (#44176)

2026-06-09 22:12:17 +08:00

layernorm.py

[Refactor] Move fla to third party (#48500)

2026-07-16 19:22:36 +01:00

lightning_attn.py

[Kernel][Test] Extend lightning_attn and awq_triton kernel tests to XPU (#43307)

2026-06-04 14:25:59 -04:00

linear.py

[CPU] Add AMX-only high-performance MLA backend for DeepSeek V2/V3/R1 (#52616)

2026-08-19 10:28:52 +08:00

logits_processor.py

[Spec Decode] DFlash2: local convolution + candidate selector (#52816)

2026-08-20 22:27:22 -07:00

mhc.py

[ROCm][Perf] Fuse DeepSeek-V4 mHC post/pre and RMSNorm with AITER (#52737)

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

mla.py

[New model] Kimi K3 (#50000)

2026-07-30 03:49:58 -07:00

resampler.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

sparse_attn_indexer.py

[ROCm][Perf] Optimize DeepSeek V4 C4A top-k with AITER (#52882)

2026-08-21 10:00:39 -07:00

utils.py

[Bugfix][CPU] Enable C++ causal_conv1d GDN path and float32 SSM cache on non-AMX AVX-512BF16 CPUs (#49688)

2026-08-19 10:50:59 -05:00

vocab_parallel_embedding.py

Fully generalise input embedding handling in Transformers modelling backend (#51247)

2026-08-06 16:21:24 +00:00