Simplify model CI and add repository review guidance

.coderabbit.yaml

@@ -6,6 +6,10 @@ reviews:
   request_changes_workflow: false
   fail_commit_status: false
 
+  pre_merge_checks:
+    docstrings:
+      mode: "off"
+
   high_level_summary: true
   high_level_summary_in_walkthrough: true
   collapse_walkthrough: true
@@ -21,3 +25,9 @@ reviews:
     base_branches: [".*"]
     auto_incremental_review: true
     auto_pause_after_reviewed_commits: 0
+
+knowledge_base:
+  code_guidelines:
+    enabled: true
+    filePatterns:
+      - "AGENTS.md"

.github/model-ci.json

@@ -0,0 +1,146 @@
+{
+  "version": 1,
+  "ignore": [
+    "**/*.md",
+    "*.md",
+    ".coderabbit.yaml",
+    ".coderabbit.yml",
+    "LICENSE",
+    "THIRD_PARTY_NOTICES.md"
+  ],
+  "groups": {
+    "native": [
+      "CMakeLists.txt",
+      "CMakePresets.json",
+      "cmake/**",
+      "common/**/*.cpp",
+      "common/**/*.h",
+      "common/*.cpp",
+      "common/*.h",
+      "common/**/CMakeLists.txt",
+      "common/CMakeLists.txt"
+    ],
+    "nemo-export": [
+      "asr/rnnt/python/tools/nemo_preprocessor_export.py",
+      "asr/rnnt/python/rnnt/__init__.py",
+      "asr/rnnt/python/rnnt/audio.py"
+    ],
+    "rnnt-runtime": [
+      "asr/rnnt/cpp/asr*",
+      "asr/rnnt/CMakeLists.txt",
+      "asr/rnnt/python/rnnt/validation/**",
+      "asr/rnnt/python/tests/**"
+    ],
+    "audio-test": [
+      "assets/sample.wav"
+    ]
+  },
+  "models": {
+    "parakeet": {
+      "export": {
+        "paths": [
+          "asr/rnnt/python/tools/export_parakeet_tdt_onnx.py",
+          "asr/rnnt/python/rnnt/parakeet_tdt/nemo_backend.py",
+          "asr/rnnt/python/rnnt/parakeet_tdt/__init__.py"
+        ],
+        "groups": [
+          "nemo-export"
+        ]
+      },
+      "runtime": {
+        "paths": [
+          "asr/rnnt/cpp/parakeet*",
+          "asr/rnnt/python/rnnt/parakeet_tdt/**"
+        ],
+        "groups": [
+          "native",
+          "rnnt-runtime",
+          "audio-test"
+        ]
+      }
+    },
+    "nemotron": {
+      "export": {
+        "paths": [
+          "asr/rnnt/python/tools/export_nemotron_onnx.py",
+          "asr/rnnt/python/rnnt/nemotron_asr/nemo_backend.py",
+          "asr/rnnt/python/rnnt/nemotron_asr/__init__.py"
+        ],
+        "groups": [
+          "nemo-export"
+        ]
+      },
+      "runtime": {
+        "paths": [
+          "asr/rnnt/cpp/nemotron*",
+          "asr/rnnt/python/rnnt/nemotron_asr/**"
+        ],
+        "groups": [
+          "native",
+          "rnnt-runtime",
+          "audio-test"
+        ]
+      }
+    },
+    "whisper": {
+      "export": {
+        "paths": [
+          "asr/whisper/model_export/**",
+          "common/model_export/**"
+        ]
+      },
+      "runtime": {
+        "paths": [
+          "asr/whisper/**"
+        ],
+        "groups": [
+          "native",
+          "audio-test"
+        ]
+      }
+    },
+    "sam2": {
+      "export": {
+        "paths": [
+          "vision/sam2/python/export_sam2_onnx.py",
+          "vision/sam2/python/modeling.py"
+        ]
+      },
+      "runtime": {
+        "paths": [
+          "vision/sam2/**",
+          "assets/sam2-*"
+        ],
+        "groups": [
+          "native"
+        ]
+      }
+    }
+  },
+  "build_only": {
+    "flux2": {
+      "targets": [
+        "din_flux2_cli"
+      ],
+      "paths": [
+        "image_gen/flux2/**"
+      ],
+      "groups": [
+        "native"
+      ],
+      "reason": "GPU inference requires model assets and a GPU runner; existing hosted CI only builds this sample."
+    },
+    "base": {
+      "targets": [
+        "din_base_onnx"
+      ],
+      "paths": [
+        "tests/base/**"
+      ],
+      "groups": [
+        "native"
+      ],
+      "reason": "Existing hosted CI builds this sample; it has no portable CPU smoke recipe."
+    }
+  }
+}

.github/model-tests/nemotron.json

@@ -0,0 +1,76 @@
+{
+  "name": "nemotron",
+  "source_roots": [
+    "asr/rnnt"
+  ],
+  "targets": [
+    "din_asr_nemotron_cli"
+  ],
+  "export": {
+    "cache_epoch": 1,
+    "install": [
+      [
+        "{python}",
+        "-m",
+        "pip",
+        "install",
+        "--index-url",
+        "https://download.pytorch.org/whl/cpu",
+        "torch"
+      ],
+      [
+        "{python}",
+        "-m",
+        "pip",
+        "install",
+        "onnx",
+        "onnxscript",
+        "soundfile",
+        "nemo_toolkit[asr] @ git+https://github.com/NVIDIA/NeMo.git@cf724ac337d1ebc7d0dda1e23fb80916f52927a5"
+      ]
+    ],
+    "command": [
+      "{python}",
+      "asr/rnnt/python/tools/export_nemotron_onnx.py",
+      "--model",
+      "{model_id}",
+      "--output",
+      "{output}",
+      "--device",
+      "cpu",
+      "--dtype",
+      "{dtype}",
+      "--external-data"
+    ],
+    "required_files": [
+      "preprocessor.onnx",
+      "encoder_step.onnx",
+      "predict_step.onnx",
+      "metadata.json",
+      "vocab.txt",
+      "decoder_init_hidden.f32",
+      "decoder_init_cell.f32"
+    ],
+    "env": {
+      "PYTHONPATH": "asr/rnnt/python"
+    }
+  },
+  "smoke": {
+    "command": [
+      "{executable}",
+      "assets/sample.wav",
+      "--model-dir",
+      "{output}",
+      "--provider",
+      "cpu"
+    ]
+  },
+  "variants": [
+    {
+      "slug": "nemotron-3.5-asr-streaming-0.6b",
+      "model_id": "nvidia/nemotron-3.5-asr-streaming-0.6b",
+      "precision": "fp16",
+      "fallback_precision": "fp32"
+    }
+  ]
+}

.github/model-tests/parakeet.json

@@ -0,0 +1,74 @@
+{
+  "name": "parakeet",
+  "source_roots": [
+    "asr/rnnt"
+  ],
+  "targets": [
+    "din_asr_parakeet_tdt_cli"
+  ],
+  "export": {
+    "cache_epoch": 1,
+    "install": [
+      [
+        "{python}",
+        "-m",
+        "pip",
+        "install",
+        "--index-url",
+        "https://download.pytorch.org/whl/cpu",
+        "torch"
+      ],
+      [
+        "{python}",
+        "-m",
+        "pip",
+        "install",
+        "onnx",
+        "onnxscript",
+        "soundfile",
+        "nemo_toolkit[asr] @ git+https://github.com/NVIDIA/NeMo.git@cf724ac337d1ebc7d0dda1e23fb80916f52927a5"
+      ]
+    ],
+    "command": [
+      "{python}",
+      "asr/rnnt/python/tools/export_parakeet_tdt_onnx.py",
+      "--model",
+      "{model_id}",
+      "--output",
+      "{output}",
+      "--device",
+      "cpu",
+      "--dtype",
+      "{dtype}",
+      "--external-data"
+    ],
+    "required_files": [
+      "preprocessor.onnx",
+      "encoder.onnx",
+      "predict_step.onnx",
+      "metadata.json",
+      "tokenizer.json"
+    ],
+    "env": {
+      "PYTHONPATH": "asr/rnnt/python"
+    }
+  },
+  "smoke": {
+    "command": [
+      "{executable}",
+      "assets/sample.wav",
+      "--model-dir",
+      "{output}",
+      "--provider",
+      "cpu"
+    ]
+  },
+  "variants": [
+    {
+      "slug": "parakeet-tdt",
+      "model_id": "nvidia/parakeet-tdt-0.6b-v3",
+      "precision": "fp16",
+      "fallback_precision": "fp32"
+    }
+  ]
+}