kubeflow
Kubeflow Python SDK to manage ML workloads and to interact with Kubeflow APIs.
Decision gist · record as of 2026-08-14
Yes. The SDK is actively maintained (release 4 days old), has no known vulnerabilities, uses a permissive Apache-2.0 license, and installs with low friction. It is the canonical way to interact with Kubeflow from Python if you are already running or planning to run Kubeflow on Kubernetes. Install it if you need to programmatically submit AI workloads to Kubeflow; skip it if you have no Kubernetes infrastructure or prefer direct kubectl/YAML workflows.AI-flagged interpretation of the facts on this page — verify before relying
Before you install
- Requires Kubernetes cluster for production use; local development backends (Docker, Podman) available via optional extras (kubeflow[docker] or kubeflow[podman]).
- Low friction install via pip with a pure-Python wheel.
- Actively maintained with a release 4 days old and recent commits; supports current Python versions (3.10, 3.11, 3.12).
License · maintenance · safety
Apache-2.0 (permissive) — Apache-2.0 permissive license allows commercial use, modification, and distribution with minimal restrictions—suitable for proprietary projects.
last release 2026-08-10 (4 days) · last repo commit 2026-08-12 · 140 stars
0 known vulnerabilities (OSV.dev, 2026-08-14) · 123,799 downloads/mo, #11,900 on PyPI
Alternatives
Verify before relying
pip install -U kubeflow
from kubeflow.trainer import TrainerClient, CustomTrainer, TrainJobTemplate
template = TrainJobTemplate(
runtime="torch-distributed",
trainer=CustomTrainer(func=train_fn, num_nodes=3)
)
job_id = TrainerClient().train(**template)- Whether kubeflow-katib-api and kubeflow-trainer-api are bundled or must be installed separately.
- Performance characteristics and scaling limits for the number of concurrent training jobs or pipeline runs.
- Compatibility matrix between Kubeflow SDK versions and specific Kubernetes cluster versions.
What it is and what it does
Kubeflow SDK is a Python library that abstracts Kubernetes complexity to let you run distributed AI workloads—training jobs, hyperparameter sweeps, model registry operations, and ML pipelines—using familiar Python APIs. It unifies access to multiple Kubeflow components (Trainer, Katib, Model Registry, Spark Operator, Pipelines) through a single SDK, so you write Python code rather than YAML manifests or kubectl commands.
The SDK supports multiple execution backends: production Kubernetes clusters, local Docker/Podman containers for development, and Python subprocesses for quick prototyping. You define training jobs, optimization experiments, or pipelines in Python, submit them via client objects, and monitor their progress. Runtime dependencies include kubernetes (for cluster interaction), pydantic (for configuration validation), and requests (for HTTP calls).
Use it for
- Submit distributed PyTorch or TensorFlow training jobs to a Kubernetes cluster without writing deployment manifests.
- Run hyperparameter optimization sweeps using Katib, specifying search spaces and trial counts in Python.
- Register and manage model versions in a Model Registry, tracking metadata and artifacts for production deployments.
- Prototype training logic locally using ContainerBackend or LocalProcessBackend before deploying to Kubernetes.
- Orchestrate multi-step ML pipelines with component definitions and run them on Kubeflow Pipelines.
- Execute Spark data processing jobs on Kubernetes via SparkClient with configurable executors and resources.
Worth the install?
AI-flagged interpretation of the facts on this page. Verify before relying on it.
Yes.
The SDK is actively maintained (release 4 days old), has no known vulnerabilities, uses a permissive Apache-2.0 license, and installs with low friction. It is the canonical way to interact with Kubeflow from Python if you are already running or planning to run Kubeflow on Kubernetes. Install it if you need to programmatically submit AI workloads to Kubeflow; skip it if you have no Kubernetes infrastructure or prefer direct kubectl/YAML workflows.
Install
kubeflow on PyPI
Before you install
Low friction install via pip with a pure-Python wheel. Actively maintained with a release 4 days old and recent commits; supports current Python versions (3.10, 3.11, 3.12).
Requires Kubernetes cluster for production use; local development backends (Docker, Podman) available via optional extras (kubeflow[docker] or kubeflow[podman]).
License in practice
Apache-2.0 permissive license allows commercial use, modification, and distribution with minimal restrictions—suitable for proprietary projects.
Quickstart
pip install -U kubeflow
from kubeflow.trainer import TrainerClient, CustomTrainer, TrainJobTemplate
template = TrainJobTemplate(
runtime="torch-distributed",
trainer=CustomTrainer(func=train_fn, num_nodes=3)
)
job_id = TrainerClient().train(**template)Verify before relying
- Whether kubeflow-katib-api and kubeflow-trainer-api are bundled or must be installed separately.
- Performance characteristics and scaling limits for the number of concurrent training jobs or pipeline runs.
- Compatibility matrix between Kubeflow SDK versions and specific Kubernetes cluster versions.
Package facts
| License | Apache-2.0 permissive |
| Python support | Supports the current Python release >=3.10 |
| Install friction | Low. Pure-Python wheel |
| Runtime dependencies | 5 packageskubeflow-katib-apikubeflow-trainer-apikubernetespydanticrequests |
| Maintenance | Actively maintained 4 days since the last release |
| Last repo commit | |
| First released | |
| Downloads | 123,799 / month, #11,900 on PyPI 30-day window, as of 2026-08-14 |
| Known vulnerabilities | None known OSV.dev, checked 2026-08-14 |
| Classifiers | Intended Audience :: DevelopersIntended Audience :: EducationIntended Audience :: Science/ResearchLicense :: OSI Approved :: Apache Software LicenseOperating System :: OS IndependentProgramming Language :: Python :: 3.10Programming Language :: Python :: 3.11Programming Language :: Python :: 3.12Topic :: Scientific/EngineeringTopic :: Scientific/Engineering :: Artificial IntelligenceTopic :: Software DevelopmentTopic :: Software Development :: LibrariesTopic :: Software Development :: Libraries :: Python Modules |
Evidence: kubeflow-0.5.0-py3-none-any.whl
Tags
Let your AI agent find packages like this
Example. Real query, live index.
You found this page by searching. An agent finds it by wishing: SkillFed indexes 14,416 PyPI packages by what they can do, searchable in plain language.
wish › “model training kubernetes”
- kubeflowKubeflow SDK provides Python APIs to submit and manage distributed AI…
- outerboundsOuterbounds is the main platform package that integrates data science…
- torchxTorchX is a universal job launcher that submits PyTorch training and…
Give your agent the search over MCP, or paste the wish link into any chat.
More Software Development packages
Provides backported and experimental type hints for Python 3.9+, allowing use of newer typing features on older Python versions and enabling early experimentation with type system PEPs before they enter the standard library.
NumPy provides an N-dimensional array object and a comprehensive suite of mathematical, linear algebra, Fourier transform, and random number functions for scientific computing in Python.
FastAPI is a Python web framework for building REST APIs using type hints, with automatic request validation, serialization, and interactive API documentation.
Provides a way to document function parameters, class attributes, return types, and variables inline using Python's `Annotated` type hint syntax instead of traditional docstrings.
Typer builds command-line applications from Python functions using type hints, automatically generating help text, argument parsing, and shell completion.
Install it if you are building CLIs in Python.
Distlib provides low-level packaging utilities for building, distributing, and managing Python software—including metadata handling, version specifiers, wheel support, script installation, and dependency resolution.
See also hopsworks · kfp · kfp-pipeline-spec · kfp-server-api · google-cloud-pipeline-components · kfp-kubernetes · torchx · orion-py-client · azureml-pipeline · kumoai