Run with Kubeflow
How to run Kueue with Kubeflow
The tasks below show you how to run Kueue managed Kubeflow Jobs.
MPI Operator Integration
Trainer Integration
Note
Kueue integration with Kubeflow Trainer supports:
- Trainer v2.0+: Use TrainJob with ClusterTrainingRuntime and TrainingRuntime
- Trainer v1.9.x and earlier: Use traditional job types (PyTorchJob, TFJob, etc.) as documented below
Trainer v2 (Recommended):
Trainer v1 (Legacy):
- Run a Kueue managed Kubeflow PyTorchJob.
- Run a Kueue managed Kubeflow TFJob.
- Run a Kueue managed Kubeflow XGBoostJob.
- Run a Kueue managed kubeflow PaddleJob.
- Run a Kueue managed kubeflow JAXJob.
Feedback
Was this page helpful?
Glad to hear it! Please tell us how we can improve.
Sorry to hear that. Please tell us how we can improve.