MLflow Kubeflow Integration

This repository provides the integration layer between MLflow and Kubeflow, making MLflow the first-class experiment tracking experience for the Kubeflow platform (KEP-897).

This repository packages two MLflow extensions for Kubernetes-backed deployments:

a workspace provider that maps MLflow workspaces to Kubernetes namespaces
an optional authorization plugin that enforces Kubernetes RBAC for MLflow requests

These plugins build on top of MLflow's workspace support. If you are new to MLflow workspaces, start with the official guide: https://mlflow.org/docs/latest/self-hosting/workspaces/getting-started/.

Components

Entry point	MLflow hook	Purpose
`kubernetes`	`mlflow.workspace_provider`	Exposes Kubernetes namespaces as MLflow workspaces.
`kubernetes-auth`	`mlflow.app`	Wraps the MLflow server with Kubernetes-based authorization checks.

Install

Install from PyPI:

pip install mlflow-kubernetes-plugins

For local development:

uv sync --extra dev

For Kubernetes deployment, see the charts/mlflow installation guide.

Quick Start

Enable MLflow workspaces on an MLflow server backed by a SQL store.
Install this package into the same environment as the MLflow server.
Configure the workspace provider and, if needed, the auth plugin.

export MLFLOW_K8S_WORKSPACE_LABEL_SELECTOR="mlflow-enabled=true"
export MLFLOW_K8S_DEFAULT_WORKSPACE="team-a"

mlflow server \
  --backend-store-uri postgresql://user:pass@localhost/mlflow \
  --default-artifact-root s3://mlflow-artifacts \
  --enable-workspaces \
  --workspace-store-uri "kubernetes://" \
  --app-name kubernetes-auth

Use --app-name kubernetes-auth only when you want request authorization enforced by Kubernetes RBAC.

Documentation

docs/index.md: docs index
docs/workspace-provider.md: workspace provider behavior, configuration, and startup
docs/authorization-plugin.md: auth modes, headers, and request handling
docs/kubernetes-rbac.md: RBAC requirements and example manifests
config/crd/bases/mlflow.kubeflow.org_mlflowconfigs.yaml: generated MLflowConfig CRD manifest

Development

See CONTRIBUTING.md for development setup, coding style, and testing instructions.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

mlflow_kubernetes_plugins-1.6.0.tar.gz (98.8 kB view details)

Uploaded Sep 2, 2026 Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

The dropdown lists show the available interpreters, ABIs, and platforms. Enable javascript to be able to filter the list of wheel files.

mlflow_kubernetes_plugins-1.6.0-py3-none-any.whl (72.3 kB view details)

Uploaded Sep 2, 2026 Python 3

File details

Details for the file mlflow_kubernetes_plugins-1.6.0.tar.gz.

File metadata

Download URL: mlflow_kubernetes_plugins-1.6.0.tar.gz
Upload date: Sep 2, 2026
Size: 98.8 kB
Tags: Source
Uploaded using Trusted Publishing? Yes
Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for mlflow_kubernetes_plugins-1.6.0.tar.gz
Algorithm	Hash digest
SHA256	`eb1b5aed5bfd81648db99ce3f796c62beeb0e6730da341bcc322baf362f429b7`
MD5	`6f3c0b007fbc646346505dbe961d4a1f`
BLAKE2b-256	`7825c3e9ce84892a7c4768e11c46e3bee7e152ecca0c0d73899a23c965bed279`

See more details on using hashes here.

Provenance

The following attestation bundles were made for mlflow_kubernetes_plugins-1.6.0.tar.gz:

Publisher: publish.yml on kubeflow/mlflow-integration

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Statement:
- Statement type: https://in-toto.io/Statement/v1
- Predicate type: https://docs.pypi.org/attestations/publish/v1
- Subject name: mlflow_kubernetes_plugins-1.6.0.tar.gz
- Subject digest: eb1b5aed5bfd81648db99ce3f796c62beeb0e6730da341bcc322baf362f429b7
- Sigstore transparency entry: 2688873465
- Sigstore integration time: Sep 2, 2026
Source repository:
- Permalink: kubeflow/mlflow-integration@26ff6d79de24f1951a2081ffa6fc0b889a1f897e
- Branch / Tag: refs/tags/v1.6.0
- Owner: https://github.com/kubeflow
- Access: public
Publication detail:
- Token Issuer: https://token.actions.githubusercontent.com
- Runner Environment: github-hosted
- Publication workflow: publish.yml@26ff6d79de24f1951a2081ffa6fc0b889a1f897e
- Trigger Event: push