mlflow-oidc-auth

MLflow auth plugin to use OpenID Connect (OIDC) as authentication and authorization provider.

This plugin allows you to use OIDC for user management in MLflow, enabling single sign-on (SSO) capabilities and centralized user management.

Features

• OIDC-based authentication for MLflow UI and API
• User management through OIDC provider
• User-level access control
• Group-based access control
• Permissions management based on regular expressions (allows or denies access to specific MLflow resources based on regular expressions and assigns permissions to users or groups)
• Support for session, JWT, and basic authentication methods
• Compatible with mlflow-client (basic auth)

Documentation

For detailed documentation, please refer to the docs. AI generated documentation is available at DeepWiki.

Quick Start

To get the full version (with entire MLflow and all dependencies), run:

python3 -m venv venv
source venv/bin/activate
python3 -m pip install mlflow-oidc-auth[full]
mlflow server --dev --app-name oidc-auth --host 0.0.0.0 --port 8080

Development

For development quick start, please refer to the Development and Contribution section.

License

Apache 2 Licensed. For more information, please see LICENSE.

Based on MLflow basic-auth plugin

https://github.com/mlflow/mlflow/tree/master/mlflow/server/auth