diff --git a/nemo_run/core/execution/slurm.py b/nemo_run/core/execution/slurm.py
index df43cd6c..294cdf1a 100644
--- a/nemo_run/core/execution/slurm.py
+++ b/nemo_run/core/execution/slurm.py
@@ -241,6 +241,7 @@ def your_slurm_executor() -> run.SlurmExecutor:
         "wait_all_nodes",
         "wckey",
         "wrap",
+        "segment",
     ]
 
     SRUN_ARGS = [
@@ -338,6 +339,7 @@ class ResourceRequest:
     monitor_group_job_wait_time: int = 60
     setup_lines: Optional[str] = None
     het_group_indices: Optional[list[int]] = None
+    segment: Optional[int] = None
 
     #: Set by the executor; cannot be initialized
     job_name: str = field(init=False, default="nemo-job")