Tracking and config of machine learning runs


Keywords
mlrun, mlops, data-science, machine-learning, experiment-tracking, data-engineering, kubernetes, mlops-workflow, model-serving, python, workflow
License
Apache-2.0
Install
pip install mlrun==1.7.0rc13

Documentation

Build Status License PyPI version fury.io Documentation Ruff GitHub commit activity GitHub release (latest SemVer) Join MLOps Live

MLRun logo

Using MLRun

MLRun is an open MLOps platform for quickly building and managing continuous ML applications across their lifecycle. MLRun integrates into your development and CI/CD environment and automates the delivery of production data, ML pipelines, and online applications, significantly reducing engineering efforts, time to production, and computation resources. With MLRun, you can choose any IDE on your local machine or on the cloud. MLRun breaks the silos between data, ML, software, and DevOps/MLOps teams, enabling collaboration and fast continuous improvements.

Get started with MLRun Tutorials and Examples, Installation and setup guide, or read about MLRun Architecture.

This page explains how MLRun addresses the MLOps Tasks and the MLRun core components.

MLOps tasks

mlrun-tasks


The MLOps development workflow section describes the different tasks and stages in detail. MLRun can be used to automate and orchestrate all the different tasks or just specific tasks (and integrate them with what you have already deployed).

Project management and CI/CD automation

In MLRun the assets, metadata, and services (data, functions, jobs, artifacts, models, secrets, etc.) are organized into projects. Projects can be imported/exported as a whole, mapped to git repositories or IDE projects (in PyCharm, VSCode, etc.), which enables versioning, collaboration, and CI/CD. Project access can be restricted to a set of users and roles.

See: Docs: Projects and Automation, CI/CD Integration, Tutorials: Quick start, Automated ML Pipeline, Video: Quick start.

Ingest and process data

MLRun provides abstract interfaces to various offline and online data sources, supports batch or realtime data processing at scale, data lineage and versioning, structured and unstructured data, and more. In addition, the MLRun Feature Store automates the collection, transformation, storage, catalog, serving, and monitoring of data features across the ML lifecycle and enables feature reuse and sharing.

See: Docs: Ingest and process data, Feature Store, Data & Artifacts; Tutorials: Quick start, Feature Store.

Develop and train models

MLRun allows you to easily build ML pipelines that take data from various sources or the Feature Store and process it, train models at scale with multiple parameters, test models, tracks each experiments, register, version and deploy models, etc. MLRun provides scalable built-in or custom model training services, integrate with any framework and can work with 3rd party training/auto-ML services. You can also bring your own pre-trained model and use it in the pipeline.

See: Docs: Develop and train models, Model Training and Tracking, Batch Runs and Workflows; Tutorials: Train, compare, and register models, Automated ML Pipeline; Video: Train and compare models.

Deploy models and applications

MLRun rapidly deploys and manages production-grade real-time or batch application pipelines using elastic and resilient serverless functions. MLRun addresses the entire ML application: intercepting application/user requests, running data processing tasks, inferencing using one or more models, driving actions, and integrating with the application logic.

See: Docs: Deploy models and applications, Realtime Pipelines, Batch Inference, Tutorials: Realtime Serving, Batch Inference, Advanced Pipeline; Video: Serving pre-trained models.

Monitor and alert

Observability is built into the different MLRun objects (data, functions, jobs, models, pipelines, etc.), eliminating the need for complex integrations and code instrumentation. With MLRun, you can observe the application/model resource usage and model behavior (drift, performance, etc.), define custom app metrics, and trigger alerts or retraining jobs.

See: Docs: Monitor and alert, Model Monitoring Overview, Tutorials: Model Monitoring & Drift Detection.

MLRun core components

mlrun-core


MLRun includes the following major components:

Project Management: A service (API, SDK, DB, UI) that manages the different project assets (data, functions, jobs, workflows, secrets, etc.) and provides central control and metadata layer.

Functions: automatically deployed software package with one or more methods and runtime-specific attributes (such as image, libraries, command, arguments, resources, etc.).

Data & Artifacts: Glueless connectivity to various data sources, metadata management, catalog, and versioning for structures/unstructured artifacts.

Feature Store: automatically collects, prepares, catalogs, and serves production data features for development (offline) and real-time (online) deployment using minimal engineering effort.

Batch Runs & Workflows: Execute one or more functions with specific parameters and collect, track, and compare all their results and artifacts.

Real-Time Serving Pipeline: Rapid deployment of scalable data and ML pipelines using real-time serverless technology, including API handling, data preparation/enrichment, model serving, ensembles, driving and measuring actions, etc.

Real-Time monitoring: monitors data, models, resources, and production components and provides a feedback loop for exploring production data, identifying drift, alerting on anomalies or data quality issues, triggering retraining jobs, measuring business impact, etc.