Add Nemotron6-MoE VLM model provider for MIMO example - #5374

Merged

yashaswikarnati merged 18 commits into

NVIDIA:mainfrom

yashaswikarnati:x1-nemotron-provider

Jun 25, 2026

Commits on Jun 23, 2026

mimo example: add Nemotron6-MoE VLM model provider
yashaswikarnati
committed
a59485f
mimo: trim verbose docstrings and prototype mentions from Nemotron provider

yashaswikarnati
and
claude
committed
f954ac2
mimo: build Nemotron VLM configs from stock args + overrides

yashaswikarnati
and
claude
committed
8895985
tests: build parity-test args via the production arg pipeline

yashaswikarnati
and
claude
committed
3eac908
mimo: pass MoE dispatcher type via CLI so the base config validates

yashaswikarnati
and
claude
committed
89889bf
tests: correct larger-variant parity case to 54 layers

yashaswikarnati
and
claude
committed
1c39400
mimo: stack Nemotron VLM provider on the RADIO encoder leaf

yashaswikarnati
and
claude
committed
7cf580f
tests: drop seq_length from the language-config parity comparison

yashaswikarnati
and
claude
committed
829a54e
mimo: trim provider to non-data args; drop the post-parse preset

yashaswikarnati
and
claude
committed
99bd416
mimo: drop vision_encoder_key publish + prepare_model_provider_args

yashaswikarnati
and
claude
committed
df324a6
mimo: drop dead predicate + hardcodes; single-source encoder name

yashaswikarnati
and
claude
committed
eef7425
mimo: trim provider docstring/comments; drop --dynamic-resolution (now a radio arg)

yashaswikarnati
and
claude
committed
f4e4a70
mimo: assert pg groups when a collection is provided (no silent fallback)

yashaswikarnati
and
claude
committed
f2ec624
mimo: direct-device init for language + projection configs

yashaswikarnati
and
claude
committed
e10b9ea
mimo: use stock dtype args for Nemotron VLM provider
yashaswikarnati
committed
38d3435
mimo: reuse core process group helpers
yashaswikarnati
committed
50ace50
mimo: simplify Nemotron provider config overrides
yashaswikarnati
committed
d6fa254

Commits on Jun 24, 2026

mimo: derive Nemotron projection input size
yashaswikarnati
committed
b4ec1ff

Navigation Menu

Uh oh!

Add Nemotron6-MoE VLM model provider for MIMO example - #5374

Add Nemotron6-MoE VLM model provider for MIMO example#5374
yashaswikarnati merged 18 commits into
NVIDIA:mainfrom
yashaswikarnati:x1-nemotron-provider

Commits on Jun 23, 2026

mimo example: add Nemotron6-MoE VLM model provider

mimo: trim verbose docstrings and prototype mentions from Nemotron provider

mimo: build Nemotron VLM configs from stock args + overrides

tests: build parity-test args via the production arg pipeline

mimo: pass MoE dispatcher type via CLI so the base config validates

tests: correct larger-variant parity case to 54 layers

mimo: stack Nemotron VLM provider on the RADIO encoder leaf

tests: drop seq_length from the language-config parity comparison

mimo: trim provider to non-data args; drop the post-parse preset

mimo: drop vision_encoder_key publish + prepare_model_provider_args

mimo: drop dead predicate + hardcodes; single-source encoder name

mimo: trim provider docstring/comments; drop --dynamic-resolution (now a radio arg)

mimo: assert pg groups when a collection is provided (no silent fallback)

mimo: direct-device init for language + projection configs

mimo: use stock dtype args for Nemotron VLM provider

mimo: reuse core process group helpers

mimo: simplify Nemotron provider config overrides

Commits on Jun 24, 2026

mimo: derive Nemotron projection input size