feat: add extra_resource_requests and extra_resource_limits to KubeflowExecutor (#479)

Signed-off-by: oliver könig <okoenig@nvidia.com>
Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>

Author: 2 people

nemo_run/core/execution/kubeflow.py

@@ -74,6 +74,8 @@ class KubeflowExecutor(Executor):
     memory_requests: Optional[str] = None
     cpu_limits: Optional[str] = None
     memory_limits: Optional[str] = None
+    extra_resource_requests: dict[str, str] = field(default_factory=dict)
+    extra_resource_limits: dict[str, str] = field(default_factory=dict)
     volume_mounts: list[dict[str, Any]] = field(default_factory=list)
     volumes: list[dict[str, Any]] = field(default_factory=list)
     labels: dict[str, Any] = field(default_factory=dict)
@@ -161,6 +163,8 @@ def _build_resources(self) -> dict[str, Any]:
         if self.gpus_per_node is not None:
             limits["nvidia.com/gpu"] = str(self.gpus_per_node)
             requests["nvidia.com/gpu"] = str(self.gpus_per_node)
+        requests.update(self.extra_resource_requests)
+        limits.update(self.extra_resource_limits)
         resources: dict[str, Any] = {}
         if limits:
             resources["limits"] = limits