Kubeflow Trainer v2.2

Repository: github.com/kubeflow/trainer
Latest release: v2.2.0 (March 20, 2026)
API version: trainer.kubeflow.org/v1alpha1
Kubernetes: v1.36
Underlying orchestration: JobSet v0.10.1

Architecture

Single TrainJob CRD replaces framework-specific CRDs. TrainingRuntime defines the execution template. Framework-agnostic: one abstraction for PyTorch, TensorFlow, JAX, XGBoost, MPI, Flux.

TrainJob (training intent)
  └─ runtimeRef ──> TrainingRuntime / ClusterTrainingRuntime (execution template)
                     └─ template ──> JobSet (underlying orchestration)
                                      └─ replicatedJobs ──> Jobs ──> Pods