Files

T

Nick Hill 234df712ed [MRV2] Minor model_runner.py code cleanup (#58610 )

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

2026-09-25 07:59:27 -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

[Bugfix][MRV2] Treat padded prompt tails as spec-decode rows for hybrid models (#58434 )

2026-09-24 18:29:53 -07:00

pool

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

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

sample

[MRV2] Validate MRV2 entrypoint logits processors (#57728 )

2026-09-24 09:07:26 +00:00

spec_decode

[SpecDecode] Add LiLiCorr drafter (#57934 )

2026-09-25 10:08:28 -04: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

[Bugfix] Capture prefill kernels for mixed FULL graphs (#58275 )

2026-09-24 02:05:07 -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

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

2026-09-24 08:48:36 -07:00

kv_connector.py

[Bugfix][KVConnector] Finalize saves on steps without a forward (#57775 )

2026-09-25 00:06:36 +00:00

lora_utils.py

Add pydocstyle to the ruff rules (#52136 )

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

model_runner.py

[MRV2] Minor model_runner.py code cleanup (#58610 )

2026-09-25 07:59:27 -07:00

pcp_manager.py

[PCP] Support prefill context parallelism with data parallelism (#57075 )

2026-09-23 22:32:36 -04:00

pp_utils.py

[PP][XPU]Add the flag to control microbatch feature on MRV2+PP (#55145 )

2026-09-25 12:05:35 +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

[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

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

2026-09-24 08:48:36 -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.