kubeflow · vsoch · Oct 26, 2025 · Jan 21, 2026 · Feb 4, 2026 · Feb 5, 2026
diff --git a/api/openapi-spec/swagger.json b/api/openapi-spec/swagger.json
diff --git a/api/python_api/kubeflow_trainer_api/models/__init__.py b/api/python_api/kubeflow_trainer_api/models/__init__.py
diff --git a/api/python_api/kubeflow_trainer_api/models/trainer_v1alpha1_flux_ml_policy_source.py b/api/python_api/kubeflow_trainer_api/models/trainer_v1alpha1_flux_ml_policy_source.py
diff --git a/api/python_api/kubeflow_trainer_api/models/trainer_v1alpha1_hpcml_policy_source.py b/api/python_api/kubeflow_trainer_api/models/trainer_v1alpha1_hpcml_policy_source.py
diff --git a/api/python_api/kubeflow_trainer_api/models/trainer_v1alpha1_ml_policy.py b/api/python_api/kubeflow_trainer_api/models/trainer_v1alpha1_ml_policy.py
diff --git a/api/python_api/kubeflow_trainer_api/models/trainer_v1alpha1_ml_policy_source.py b/api/python_api/kubeflow_trainer_api/models/trainer_v1alpha1_ml_policy_source.py
diff --git a/charts/kubeflow-trainer/crds/trainer.kubeflow.org_clustertrainingruntimes.yaml b/charts/kubeflow-trainer/crds/trainer.kubeflow.org_clustertrainingruntimes.yaml
@@ -47,6 +47,19 @@ spec:
                 description: mlPolicy provides the ML-specific parameters for the
                   model training.
                 properties:
+                  flux:
+                    description: flux defines the configuration for the Flux runtime.
+                    properties:
+                      numProcPerNode:
+                        default: 1
+                        description: numProcPerNode is the number of processes per
+                          node.
+                        format: int32
+                        type: integer
+                        x-kubernetes-validations:
+                        - message: NumProcPerNode in fluxPolicy must be >= 1
+                          rule: self >= 1
+                    type: object
                   mpi:
                     description: mpi defines the configuration for the MPI Runtime.
                     properties:

diff --git a/charts/kubeflow-trainer/crds/trainer.kubeflow.org_trainingruntimes.yaml b/charts/kubeflow-trainer/crds/trainer.kubeflow.org_trainingruntimes.yaml
@@ -47,6 +47,19 @@ spec:
                 description: mlPolicy provides the ML-specific parameters for the
                   model training.
                 properties:
+                  flux:
+                    description: flux defines the configuration for the Flux runtime.
+                    properties:
+                      numProcPerNode:
+                        default: 1
+                        description: numProcPerNode is the number of processes per
+                          node.
+                        format: int32
+                        type: integer
+                        x-kubernetes-validations:
+                        - message: NumProcPerNode in fluxPolicy must be >= 1
+                          rule: self >= 1
+                    type: object
                   mpi:
                     description: mpi defines the configuration for the MPI Runtime.
                     properties:

diff --git a/examples/flux/flux-runtime.yaml b/examples/flux/flux-runtime.yaml
@@ -0,0 +1,25 @@
+apiVersion: trainer.kubeflow.org/v1alpha1
+kind: ClusterTrainingRuntime
+metadata:
+  name: flux-runtime
+  labels:
+    trainer.kubeflow.org/framework: flux
+spec:
+  mlPolicy:
+    numNodes: 1
+    flux: {}
+  template:
+    spec:
+      replicatedJobs:
+        - name: node
+          template:
+            metadata:
+              labels:
+                trainer.kubeflow.org/trainjob-ancestor-step: trainer
+            spec:
+              template:
+                spec:
+                  # An init container will be added here to install flux
+                  containers:
+                    - name: node
+                      image: ubuntu:jammy