Add deprecation notice to modules

nemo/collections/avlm/__init__.py

@@ -25,3 +25,12 @@
     "AVLMModel",
     "avlm_8b",
 ]
+
+
+import warnings
+
+warnings.warn(
+    "nemo.collections.avlm is deprecated and will be removed in the next major NeMo FW container release.",
+    DeprecationWarning,
+    stacklevel=2,
+)

nemo/collections/diffusion/__init__.py

@@ -11,3 +11,12 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+import warnings
+
+
+warnings.warn(
+    "nemo.collections.diffusion is deprecated and will be removed in the next major NeMo FW container release."
+    DeprecationWarning,
+    stacklevel=2,
+)

nemo/collections/llm/__init__.py

@@ -449,3 +449,13 @@
     __all__.append("evaluate")
 except ImportError as error:
     logging.warning(f"The evaluate module could not be imported: {error}")
+
+
+import warnings
+
+
+warnings.warn(
+    "nemo.collections.llm is deprecated and will be removed in the next major NeMo FW container release.",
+    DeprecationWarning,
+    stacklevel=2,
+)

nemo/collections/multimodal/__init__.py

@@ -11,3 +11,11 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+import warnings
+
+warnings.warn(
+    "nemo.collections.multimodal is deprecated and will be removed in the next major NeMo FW container release.",
+    DeprecationWarning,
+    stacklevel=2,
+)

nemo/collections/multimodal_autoregressive/__init__.py

@@ -0,0 +1,21 @@
+# Copyright (c) 2025, NVIDIA CORPORATION.  All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import warnings
+
+warnings.warn(
+    "nemo.collections.multimodal_autoregressive is deprecated and will be removed in the next major NeMo FW container release.",
+    DeprecationWarning,
+    stacklevel=2,
+)

nemo/collections/nlp/__init__.py

@@ -25,3 +25,11 @@
 
 # Set collection name.
 __description__ = "Natural Language Processing collection"
+
+import warnings
+
+warnings.warn(
+    "nemo.collections.nlp is deprecated and will be removed in the next major NeMo FW container release.",
+    DeprecationWarning,
+    stacklevel=2,
+)

nemo/collections/vision/__init__.py

@@ -23,3 +23,11 @@
 
 # Set collection name.
 __description__ = "Computer Vision collection"
+
+import warnings
+
+warnings.warn(
+    "nemo.collections.nlp is deprecated and will be removed in the next major NeMo FW container release.",
+    DeprecationWarning,
+    stacklevel=2,
+)

nemo/collections/vlm/__init__.py

@@ -190,3 +190,11 @@
     from nemo.utils import logging
 
     logging.warning(f"Failed to import nemo.collections.vlm.api: {error}")
+
+import warnings
+
+warnings.warn(
+    "nemo.collections.vlm is deprecated and will be removed in the next major NeMo FW container release.",
+    DeprecationWarning,
+    stacklevel=2,
+)

nemo/lightning/__init__.py

@@ -78,3 +78,11 @@ def _is_slurm_interactive_mode():
     "get_vocab_size",
     "teardown",
 ]
+
+import warnings
+
+warnings.warn(
+    "nemo.lightning is deprecated and will be removed in the next major NeMo FW container release.",
+    DeprecationWarning,
+    stacklevel=2,
+)