Files

T

Benjamin Chislett 8b346309a5 [Refactor] Consolidate SupportsEagle (#36063 )

Signed-off-by: Benjamin Chislett <bchislett@nvidia.com>

2026-03-13 23:22:40 +00:00

metrics

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

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

[Misc] Rename group_mm_kwargs_by_modality -> group_and_batch_mm_kwargs (#36158 )

2026-03-06 12:32:48 +08:00

model_states

[Model Runner V2] Add WhisperModelState [6/N] (#35790 )

2026-03-11 14:20:38 -07:00

pool

[Model Runner V2] Do not initialize sampler for non-last PP ranks (#36824 )

2026-03-12 03:55:28 +00:00

sample

[Model Runner V2] Some code simplification (#36929 )

2026-03-13 00:59:23 +00:00

spec_decode

[Refactor] Consolidate SupportsEagle (#36063 )

2026-03-13 23:22:40 +00:00

__init__.py

GPU Model Runner V2 (#25266 )

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

async_utils.py

[Model Runner V2] Fix pooling (#36019 )

2026-03-04 10:53:17 -08:00

attn_utils.py

[Model Runner V2] Add WhisperModelState [6/N] (#35790 )

2026-03-11 14:20:38 -07:00

block_table.py

[Model Runner V2] Fix _compute_slot_mappings_kernel for chunked prefill (#36580 )

2026-03-10 00:23:42 -07:00

buffer_utils.py

[Model Runner V2] Misc code simplification (#35941 )

2026-03-04 15:26:35 -08:00

cp_utils.py

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

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

cudagraph_utils.py

[Model Runner V2] Add WhisperModelState [6/N] (#35790 )

2026-03-11 14:20:38 -07:00

dp_utils.py

[MRV2] Extensible CG dispatch rework (#35959 )

2026-03-09 13:58:45 -07:00

input_batch.py

[Model Runner V2] Do not initialize sampler for non-last PP ranks (#36824 )

2026-03-12 03:55:28 +00:00

kv_connector.py

[Spec Decode] Defer clearing KV connector metadata for EAGLE3 speculative decode + prefill / decode disagg setup (#34529 )

2026-02-22 08:08:32 -08:00

lora_utils.py

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

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

model_runner.py

[Model Runner V2] Do not initialize sampler for non-last PP ranks (#36824 )

2026-03-12 03:55:28 +00:00

pp_utils.py

[Model Runner V2] A bit more PP simplification (#34766 )

2026-02-17 21:39:07 -08:00

README.md

GPU Model Runner V2 (#25266 )

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

states.py

[Model Runner V2] Add probabilistic rejection sampling for spec decoding (#35461 )

2026-03-11 14:04:32 -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] Fix warmup for pipeline parallel (#36280 )

2026-03-06 16:58:51 -08: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.