Files

T

Nick Hill 310f15d354 [Bugfix][MRV2] Align dummy idx_mapping dtype to avoid runtime jit (#58462 )

Signed-off-by: Nick Hill <nickhill123@gmail.com>

2026-09-23 16:54:48 -07:00

metrics

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

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

Add pydocstyle to the ruff rules (#52136 )

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

model_states

[MRV2] Miscellaneous code cleanup (#57980 )

2026-09-23 01:51:52 -07:00

pool

[Pooling] Support prompt embeddings in MRV2 decoder pooling (#56639 )

2026-09-13 05:49:45 +00:00

sample

[Refactor] Remove dead code multiple places (#58002 )

2026-09-22 16:36:03 -04:00

spec_decode

[DSpark] Support pipeline-parallel targets in aggregated serving (#56956 )

2026-09-23 04:52:28 -07:00

__init__.py

GPU Model Runner V2 (#25266 )

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

async_utils.py

[AuxOutput] Add block-keyed storage for routed-expert outputs (#45635 )

2026-09-21 10:25:05 +00:00

attn_utils.py

[Bugfix][MRV2] Match fast-prefill padding to active LoRA batches (#56456 )

2026-09-18 22:13:23 +00:00

block_table.py

[MRV2] Miscellaneous code cleanup (#57980 )

2026-09-23 01:51:52 -07:00

buffer_utils.py

[MRV2] Buffer util simplifications (#56888 )

2026-09-15 20:36:51 +00:00

cp_utils.py

[MRV2] Miscellaneous code cleanup (#57980 )

2026-09-23 01:51:52 -07:00

cudagraph_utils.py

[MRV2] Miscellaneous code cleanup (#57980 )

2026-09-23 01:51:52 -07:00

dp_utils.py

Add pydocstyle to the ruff rules (#52136 )

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

ec_connector.py

[MRV2] Miscellaneous code cleanup (#57980 )

2026-09-23 01:51:52 -07:00

eplb_utils.py

[MRV2] Miscellaneous code cleanup (#57980 )

2026-09-23 01:51:52 -07:00

input_batch.py

[Bugfix][MRV2] Align dummy idx_mapping dtype to avoid runtime jit (#58462 )

2026-09-23 16:54:48 -07:00

kv_connector.py

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

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

lora_utils.py

Add pydocstyle to the ruff rules (#52136 )

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

model_runner.py

[CI][Bugfix] Limit MRV2 sampler JIT warmup registration to ROCm (#58465 )

2026-09-23 16:11:50 -07:00

pcp_manager.py

[MRV2] Miscellaneous code cleanup (#57980 )

2026-09-23 01:51:52 -07:00

pp_utils.py

[DSpark] Support pipeline-parallel targets in aggregated serving (#56956 )

2026-09-23 04:52:28 -07: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

[MRV2] Miscellaneous code cleanup (#57980 )

2026-09-23 01:51:52 -07:00

structured_outputs.py

[MRV2] Buffer util simplifications (#56888 )

2026-09-15 20:36:51 +00:00

ubatch_utils.py

[MRV2] Miscellaneous code cleanup (#57980 )

2026-09-23 01:51:52 -07:00

warmup.py

[DSpark] Support pipeline-parallel targets in aggregated serving (#56956 )

2026-09-23 04:52:28 -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.