Walkthrough

CI matrix updated to include llm_qat and its separate failing job removed; warning text in torch init revised; a test replaced an in-place parameter sort with a shared utility; docs gained an Advanced NAS section; a star-import of megatron_model_config from a plugin init was removed. Public export set changed.

Changes

Sequence Diagram(s)

sequenceDiagram
    autonumber
    participant Test as Test (unit)
    participant Model as DynamicSpace / Model
    participant Util as mtn.utils.sort_parameters

    Note over Test,Model: Before -> in-place call (old)
    Test->>Model: dynamic_space.sort_parameters()
    Model-->>Test: parameters sorted

    Note over Test,Util: After -> centralized utility (new)
    Test->>Util: mtn.utils.sort_parameters(model)
    Util-->>Test: parameters sorted

sequenceDiagram
    autonumber
    participant Importer as plugin __init__
    participant MegatronConfig as megntron_model_config
    participant Plugin as megatron plugin

    Note over Importer,MegatronConfig: Old (before)
    Importer->>MegatronConfig: from .megatron_model_config import *
    MegatronConfig-->>Importer: symbols re-exported

    Note over Importer,Plugin: New (after)
    Importer->>Plugin: plugin init without star-import
    Plugin-->>Importer: no direct re-exports from megatron_model_config

Estimated code review effort

🎯 3 (Moderate) | ⏱️ ~20 minutes

Poem

I twitch my whiskers at CI's new beat,
QAT hops in, no failing seat.
A gentler warning, calmly said,
Params sorted now by a shared thread.
I twitch, I nibble—changes neat. 🥕

Pre-merge checks and finishing touches

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

_{Comment @coderabbitai help to get the list of available commands and usage tips.}

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 73.79%. Comparing base (e09d36c) to head (4e1067a).
⚠️ Report is 3 commits behind head on main.

@@           Coverage Diff           @@
##             main     #412   +/-   ##
=======================================
  Coverage   73.79%   73.79%           
=======================================
  Files         171      171           
  Lines       17591    17591           
=======================================
  Hits        12982    12982           
  Misses       4609     4609           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.