Files

T

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

Signed-off-by: Giancarlo Delfin <gdelfin@inferact.ai>

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

metrics

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

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

[MRV2] Add encoder cache profiling implementation (#47985 )

2026-07-20 20:18:14 -07:00

model_states

[ModelRunner V2] Enable sequence pooling for embedding and classification models (#48791 )

2026-07-29 06:30:45 -07:00

pool

[ModelRunner V2] Enable sequence pooling for embedding and classification models (#48791 )

2026-07-29 06:30:45 -07:00

sample

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

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

spec_decode

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

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

__init__.py

GPU Model Runner V2 (#25266 )

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

async_utils.py

[ModelRunner V2] Enable sequence pooling for embedding and classification models (#48791 )

2026-07-29 06:30:45 -07:00

attn_utils.py

[ModelRunner V2] Support encoder-only attention (#49331 )

2026-07-27 00:38:44 +00:00

block_table.py

[BugFix] Stop dummy runs from writing mamba state through stale block-table rows (#49757 )

2026-07-29 01:09:41 +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

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

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

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

pcp_manager.py

[Core] Add MRV2 virtual-batch PCP for MLA (#46570 )

2026-07-19 02:53:15 +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

[ModelRunner V2] Enable sequence pooling for embedding and classification models (#48791 )

2026-07-29 06:30:45 -07: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.