Home Knowledge Base Kubeflow

Kubeflow is the cloud-native machine learning toolkit for Kubernetes that provides standardized components for ML pipelines, model serving, and notebook management — enabling organizations running Kubernetes to orchestrate ML workflows (data prep → training → evaluation → serving) as containerized pipeline steps with the Kubeflow Pipelines engine and serve models at scale with KServe.

What Is Kubeflow?

Why Kubeflow Matters for AI

Kubeflow Core Components

Kubeflow Pipelines (KFP): from kfp import dsl, compiler

@dsl.component(base_image="python:3.11", packages_to_install=["scikit-learn", "pandas"]) def preprocess(raw_data_path: str, output_path: dsl.Output[dsl.Dataset]): import pandas as pd df = pd.read_csv(raw_data_path) df_clean = df.dropna() df_clean.to_csv(output_path.path, index=False)

@dsl.component(base_image="pytorch/pytorch:2.0-cuda11.7-cudnn8-runtime") def train_model( dataset: dsl.Input[dsl.Dataset], model_output: dsl.Output[dsl.Model], learning_rate: float = 0.001 ): # Training code — runs in isolated Pod on GPU node model = train(dataset.path, lr=learning_rate) model.save(model_output.path)

@dsl.pipeline(name="ml-training-pipeline") def training_pipeline(raw_data: str, lr: float = 0.001): preprocess_task = preprocess(raw_data_path=raw_data) train_task = train_model( dataset=preprocess_task.outputs["output_path"], learning_rate=lr ).set_accelerator_type("NVIDIA_TESLA_A100").set_gpu_limit(1)

compiler.Compiler().compile(training_pipeline, "pipeline.yaml")

Kubeflow Training Operator:

KServe (Model Serving): apiVersion: serving.kserve.io/v1beta1 kind: InferenceService metadata: name: llama-server spec: predictor: model: modelFormat: name: pytorch storageUri: "s3://models/llama-3-8b" resources: limits: nvidia.com/gpu: "1" # Autoscales from 0 to 10 replicas based on traffic

Kubeflow Notebooks:

Kubeflow Operational Complexity

What It Requires:

Managed Alternatives:

Kubeflow vs Alternatives

ToolK8s RequiredSetup ComplexityGPU SupportBest For
KubeflowYesVery HighExcellentK8s-native orgs
AirflowOptionalHighVia operatorsComplex ETL + ML
PrefectOptionalLowVia K8s workerPython-first teams
Vertex AINoLowManagedGoogle Cloud users
SageMakerNoMediumManagedAWS users

Kubeflow is the Kubernetes-native ML platform for organizations that need deep cloud-infrastructure integration for their AI workflows — by expressing every ML step as a Kubernetes-native resource with containerized execution, Kubeflow enables teams already invested in Kubernetes to run reproducible, scalable ML pipelines without adopting a separate orchestration system outside their existing infrastructure.

kubeflowkubernetesml

Explore 500+ Semiconductor & AI Topics

From EUV lithography to CUDA optimization — search the full knowledge base or chat with our AI assistant.