[feat]: added one-logger support for NeMo RL

.gitmodules

@@ -18,3 +18,6 @@
 	url = https://github.com/NVIDIA-NeMo/Gym.git
 	branch = main
 	shallow = true
+[submodule "3rdparty/NV-One-Logger-workspace/nv-one-logger"]
+	path = 3rdparty/NV-One-Logger-workspace/nv-one-logger
+	url = [email protected]:NVIDIA/nv-one-logger.git

3rdparty/NV-One-Logger-workspace/is_nvlogger_installed.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.
+try:
+    import nv_one_logger # noqa: F401
+
+    INSTALLED = True
+except ImportError:
+    INSTALLED = False
+
+print(f"NV One Logger {INSTALLED=}")

3rdparty/NV-One-Logger-workspace/pyproject.toml

@@ -0,0 +1,9 @@
+[build-system]
+requires = ["setuptools", "wheel", "toml"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "nv-one-logger-core"
+description = "Wrapper for nv-one-logger-core"
+requires-python = ">=3.8"
+dynamic = ["version", "dependencies"]

3rdparty/NV-One-Logger-workspace/setup.py

@@ -0,0 +1,59 @@
+import sys
+import toml
+from pathlib import Path
+import setuptools
+
+# Define the source directory for the actual package (git-cloned repo)
+src_dir = Path(__file__).parent / "nv-one-logger" / "nv_one_logger" / "one_logger_core"
+
+if not src_dir.exists():
+    raise FileNotFoundError(f"{src_dir} not found.")
+
+# Read the underlying pyproject.toml
+pyproject_toml_path = src_dir / "pyproject.toml"
+with pyproject_toml_path.open("r") as f:
+    pyproject_data = toml.load(f)
+
+# Extract dependencies
+# Handle poetry dependencies format
+poetry_deps = pyproject_data.get("tool", {}).get("poetry", {}).get("dependencies", {})
+install_requires = []
+for pkg, ver in poetry_deps.items():
+    if pkg == "python":
+        continue
+    # Simple conversion, might need more robust handling for complex version strings
+    # but sufficient for this specific repo's needs
+    if isinstance(ver, str):
+        if ver.startswith("^"):
+            install_requires.append(f"{pkg}>={ver[1:]}")
+        else:
+            install_requires.append(f"{pkg}=={ver}")
+    elif isinstance(ver, dict):
+        # Handle cases like {version = "...", markers = "..."} if present
+        if "version" in ver:
+            version_spec = ver['version']
+            # If version already has operators (>=, ==, etc.), use as-is
+            if any(op in version_spec for op in ['>=', '<=', '==', '!=', '~=', '>', '<']):
+                install_requires.append(f"{pkg}{version_spec}")
+            else:
+                install_requires.append(f"{pkg}>={version_spec}")
+
+# Extract version
+version = pyproject_data.get("tool", {}).get("poetry", {}).get("version", "0.0.0")
+
+# Map the package directory
+# The package name in source is 'nv_one_logger' (from src directory in one_logger_core)
+# one_logger_core/src/nv_one_logger
+package_dir = {
+    "": str(src_dir / "src")
+}
+packages = setuptools.find_packages(where=str(src_dir / "src"))
+
+setuptools.setup(
+    name="nv-one-logger-core",
+    version=version,
+    description="Wrapper for nv-one-logger-core",
+    packages=packages,
+    package_dir=package_dir,
+    install_requires=install_requires,
+)

3rdparty/NV-One-Logger-workspace/wrappers/nv-one-logger-otel/pyproject.toml

@@ -0,0 +1,9 @@
+[build-system]
+requires = ["setuptools", "wheel", "toml"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "nv-one-logger-otel"
+description = "Wrapper for nv-one-logger-otel"
+requires-python = ">=3.8"
+dynamic = ["version", "dependencies"]

3rdparty/NV-One-Logger-workspace/wrappers/nv-one-logger-otel/setup.py

@@ -0,0 +1,59 @@
+import sys
+import toml
+from pathlib import Path
+import setuptools
+
+# Define the source directory for the actual package (git-cloned repo)
+src_dir = Path(__file__).parent.parent.parent / "nv-one-logger" / "nv_one_logger" / "one_logger_otel"
+
+if not src_dir.exists():
+    raise FileNotFoundError(f"{src_dir} not found.")
+
+# Read the underlying pyproject.toml
+pyproject_toml_path = src_dir / "pyproject.toml"
+with pyproject_toml_path.open("r") as f:
+    pyproject_data = toml.load(f)
+
+# Extract dependencies
+# Handle poetry dependencies format
+poetry_deps = pyproject_data.get("tool", {}).get("poetry", {}).get("dependencies", {})
+install_requires = []
+for pkg, ver in poetry_deps.items():
+    if pkg == "python":
+        continue
+    # Simple conversion, might need more robust handling for complex version strings
+    # but sufficient for this specific repo's needs
+    if isinstance(ver, str):
+        if ver.startswith("^"):
+            install_requires.append(f"{pkg}>={ver[1:]}")
+        else:
+            install_requires.append(f"{pkg}=={ver}")
+    elif isinstance(ver, dict):
+        # Handle cases like {version = "...", markers = "..."} if present
+        if "version" in ver:
+            version_spec = ver['version']
+            # If version already has operators (>=, ==, etc.), use as-is
+            if any(op in version_spec for op in ['>=', '<=', '==', '!=', '~=', '>', '<']):
+                install_requires.append(f"{pkg}{version_spec}")
+            else:
+                install_requires.append(f"{pkg}>={version_spec}")
+
+# Extract version
+version = pyproject_data.get("tool", {}).get("poetry", {}).get("version", "0.0.0")
+
+# Map the package directory
+# The package name in source is 'nv_one_logger' (from src directory in one_logger_otel)
+# one_logger_otel/src/nv_one_logger
+package_dir = {
+    "": str(src_dir / "src")
+}
+packages = setuptools.find_packages(where=str(src_dir / "src"))
+
+setuptools.setup(
+    name="nv-one-logger-otel",
+    version=version,
+    description="Extensions to onelogger library to use Open telemetry (OTEL) as a backend.",
+    packages=packages,
+    package_dir=package_dir,
+    install_requires=install_requires,
+)

3rdparty/NV-One-Logger-workspace/wrappers/nv-one-logger-training-telemetry/pyproject.toml

@@ -0,0 +1,9 @@
+[build-system]
+requires = ["setuptools", "wheel", "toml"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "nv-one-logger-training-telemetry"
+description = "Wrapper for nv-one-logger-training-telemetry"
+requires-python = ">=3.8"
+dynamic = ["version", "dependencies"]

3rdparty/NV-One-Logger-workspace/wrappers/nv-one-logger-training-telemetry/setup.py

@@ -0,0 +1,59 @@
+import sys
+import toml
+from pathlib import Path
+import setuptools
+
+# Define the source directory for the actual package (git-cloned repo)
+src_dir = Path(__file__).parent.parent.parent / "nv-one-logger" / "nv_one_logger" / "one_logger_training_telemetry"
+
+if not src_dir.exists():
+    raise FileNotFoundError(f"{src_dir} not found.")
+
+# Read the underlying pyproject.toml
+pyproject_toml_path = src_dir / "pyproject.toml"
+with pyproject_toml_path.open("r") as f:
+    pyproject_data = toml.load(f)
+
+# Extract dependencies
+# Handle poetry dependencies format
+poetry_deps = pyproject_data.get("tool", {}).get("poetry", {}).get("dependencies", {})
+install_requires = []
+for pkg, ver in poetry_deps.items():
+    if pkg == "python":
+        continue
+    # Simple conversion, might need more robust handling for complex version strings
+    # but sufficient for this specific repo's needs
+    if isinstance(ver, str):
+        if ver.startswith("^"):
+            install_requires.append(f"{pkg}>={ver[1:]}")
+        else:
+            install_requires.append(f"{pkg}=={ver}")
+    elif isinstance(ver, dict):
+        # Handle cases like {version = "...", markers = "..."} if present
+        if "version" in ver:
+            version_spec = ver['version']
+            # If version already has operators (>=, ==, etc.), use as-is
+            if any(op in version_spec for op in ['>=', '<=', '==', '!=', '~=', '>', '<']):
+                install_requires.append(f"{pkg}{version_spec}")
+            else:
+                install_requires.append(f"{pkg}>={version_spec}")
+
+# Extract version
+version = pyproject_data.get("tool", {}).get("poetry", {}).get("version", "0.0.0")
+
+# Map the package directory
+# The package name in source is 'nv_one_logger' (from src directory in one_logger_training_telemetry)
+# one_logger_training_telemetry/src/nv_one_logger
+package_dir = {
+    "": str(src_dir / "src")
+}
+packages = setuptools.find_packages(where=str(src_dir / "src"))
+
+setuptools.setup(
+    name="nv-one-logger-training-telemetry",
+    version=version,
+    description="Training job telemetry using OneLogger library.",
+    packages=packages,
+    package_dir=package_dir,
+    install_requires=install_requires,
+)

3rdparty/NV-One-Logger-workspace/wrappers/nv-one-logger-wandb/pyproject.toml

@@ -0,0 +1,9 @@
+[build-system]
+requires = ["setuptools", "wheel", "toml"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "nv-one-logger-wandb"
+description = "Wrapper for nv-one-logger-wandb"
+requires-python = ">=3.9"
+dynamic = ["version", "dependencies"]

3rdparty/NV-One-Logger-workspace/wrappers/nv-one-logger-wandb/setup.py

@@ -0,0 +1,59 @@
+import sys
+import toml
+from pathlib import Path
+import setuptools
+
+# Define the source directory for the actual package (git-cloned repo)
+src_dir = Path(__file__).parent.parent.parent / "nv-one-logger" / "nv_one_logger" / "one_logger_wandb"
+
+if not src_dir.exists():
+    raise FileNotFoundError(f"{src_dir} not found.")
+
+# Read the underlying pyproject.toml
+pyproject_toml_path = src_dir / "pyproject.toml"
+with pyproject_toml_path.open("r") as f:
+    pyproject_data = toml.load(f)
+
+# Extract dependencies
+# Handle poetry dependencies format
+poetry_deps = pyproject_data.get("tool", {}).get("poetry", {}).get("dependencies", {})
+install_requires = []
+for pkg, ver in poetry_deps.items():
+    if pkg == "python":
+        continue
+    # Simple conversion, might need more robust handling for complex version strings
+    # but sufficient for this specific repo's needs
+    if isinstance(ver, str):
+        if ver.startswith("^"):
+            install_requires.append(f"{pkg}>={ver[1:]}")
+        else:
+            install_requires.append(f"{pkg}=={ver}")
+    elif isinstance(ver, dict):
+        # Handle cases like {version = "...", markers = "..."} if present
+        if "version" in ver:
+            version_spec = ver['version']
+            # If version already has operators (>=, ==, etc.), use as-is
+            if any(op in version_spec for op in ['>=', '<=', '==', '!=', '~=', '>', '<']):
+                install_requires.append(f"{pkg}{version_spec}")
+            else:
+                install_requires.append(f"{pkg}>={version_spec}")
+
+# Extract version
+version = pyproject_data.get("tool", {}).get("poetry", {}).get("version", "0.0.0")
+
+# Map the package directory
+# The package name in source is 'nv_one_logger' (from src directory in one_logger_wandb)
+# one_logger_wandb/src/nv_one_logger
+package_dir = {
+    "": str(src_dir / "src")
+}
+packages = setuptools.find_packages(where=str(src_dir / "src"))
+
+setuptools.setup(
+    name="nv-one-logger-wandb",
+    version=version,
+    description="Extensions to onelogger library to use Weights and Biases as a backend.",
+    packages=packages,
+    package_dir=package_dir,
+    install_requires=install_requires,
+)

examples/configs/grpo_math_1B.yaml

@@ -283,6 +283,10 @@ logger:
   swanlab:
     project: "grpo-dev"
     name: "grpo-dev-logger"
+  one_logger_enabled: false
+  one_logger:
+    project: "grpo-dev"
+    name: "grpo-dev-logger"
   tensorboard: {}
   mlflow:
     experiment_name: "grpo-dev"

examples/configs/sft.yaml

@@ -206,6 +206,10 @@ logger:
   swanlab:
     project: "sft-dev"
     name: "sft-dev-${data.dataset_name}"
+  one_logger_enabled: false
+  one_logger:
+    project: "sft-dev"
+    name: "sft-dev-${data.dataset_name}"
   tensorboard:
     log_dir: "tb_logs-sft-dev-${data.dataset_name}"
   mlflow: