Files

T

gabriel-peracioandKimi K3 47a4e410ba [Bugfix][Spec Decode] Fix NaN handling in rejection sampler tl.argmax (#50183 )

Signed-off-by: Gabriel Peracio <gabriel@sigsonic.com>
Co-authored-by: Kimi K3 <noreply@moonshot.ai>

2026-08-05 18:01:09 -07:00

metrics

[Model Runner V2] Support num NaNs in logits (#30187 )

2025-12-09 10:00:49 -08:00

[Model Runner v2] E/P/D disaggregation support (#38390 )

2026-08-04 15:52:23 +00:00

model_states

[ModelRunnerV2] Fix scalar Mamba state update with int32 mappings (#50327 )

2026-08-03 22:05:12 +00:00

pool

[Model Runner v2] Enable BGE M3 pooling embed token_classify (#50661 )

2026-08-02 12:46:03 -04:00

sample

[MRV2][Performance] Skip no-op FP32 logits materialization (#47711 )

2026-07-27 15:31:36 +00:00

spec_decode

[Bugfix][Spec Decode] Fix NaN handling in rejection sampler tl.argmax (#50183 )

2026-08-05 18:01:09 -07:00

__init__.py

GPU Model Runner V2 (#25266 )

2025-11-21 08:20:55 -08:00

async_utils.py

[CI] Detect and fail evals on when NaNs appear in logits (#50323 )

2026-08-04 23:02:58 +00:00

attn_utils.py

[Bugfix][TurboQuant] Add KV quant mode for turboquant (#50533 )

2026-07-31 17:42:32 +00:00

block_table.py

[Model Runner v2] E/P/D disaggregation support (#38390 )

2026-08-04 15:52:23 +00:00

buffer_utils.py

[Core] Ensure memory is pinned prior to async h2d copy (#45424 )

2026-06-20 20:02:24 -07:00

cp_utils.py

[Model Runner V2] Minor simplification for DCP (#34786 )

2026-02-18 11:04:53 -08:00

cudagraph_utils.py

[MRV2] Always build attn metadata at capture time (#49364 ) (#49995 )

2026-07-27 17:25:17 +00:00

dp_utils.py

[Model Runner V2][Bugfix] Fix MRV2 LoRA warmup (#35536 )

2026-06-15 13:17:23 -07:00

ec_connector.py

[Model Runner v2] E/P/D disaggregation support (#38390 )

2026-08-04 15:52:23 +00:00

eplb_utils.py

[EPLB] Mask padding in EPLB load recording (#38128 )

2026-06-28 19:43:58 -07:00

input_batch.py

[Model Runner V2][Spec Decode] Add multi-layer MTP speculator (#48892 )

2026-07-30 15:54:15 -07:00

kv_connector.py

[Core] Cleanup KVConnector handling with PP + fix MRV2 (#43732 )

2026-05-28 13:12:03 -07:00

lora_utils.py

[BugFix][MRV2] Don't create dummy requests longer than max_model_len (#49751 )

2026-07-27 02:04:20 +00:00

model_runner.py

[Refactor][PCP] Make PCPManager construction extensible (#50066 )

2026-08-05 23:05:14 +00:00

pcp_manager.py

[Refactor][PCP] Make PCPManager construction extensible (#50066 )

2026-08-05 23:05:14 +00:00

pp_utils.py

[XPU] Fix PP accuracy on XPU device (#47253 )

2026-07-07 09:17:27 +08:00

README.md

GPU Model Runner V2 (#25266 )

2025-11-21 08:20:55 -08:00

shutdown.py

[MRV2] Add shutdown() method (#41297 )

2026-05-02 21:06:30 -07:00

states.py

[Model Runner V2][Spec Decode] Add multi-layer MTP speculator (#48892 )

2026-07-30 15:54:15 -07:00

structured_outputs.py

[ModelRunner V2] Misc minor simplifications and optimizations (#33467 )

2026-02-01 22:17:14 +00:00

warmup.py

[Model Runner v2] E/P/D disaggregation support (#38390 )

2026-08-04 15:52:23 +00:00

README.md

[Experimental] Model Runner V2

This directory contains the new model runner which is under active development. Ping Woosuk Kwon for any changes.