Data / ML · MLOps

MLflow Model Tracking — A Field Guide

MLflow is the common open-source backbone for ML lifecycle management. Its four components solve different problems — knowing which does what is how you use it without over-adopting.

John Kihiu12 min read

MLflow is one of the most widely used open-source platforms for managing the machine-learning lifecycle, and it is really four components that solve distinct problems. Teams often adopt one part and ignore the rest, or reach for all of it when they need a fraction. Understanding what each piece does lets you use MLflow for exactly what you need without over-adopting the whole platform.

The four components

ComponentSolves
TrackingLogging runs — parameters, metrics, artifacts — and comparing them
Model RegistryVersioning models and managing their stage (staging, production)
ModelsA standard packaging format so a model can be served consistently
ProjectsPackaging code for reproducible runs

Most teams start with Tracking, add the Model Registry when they need to manage models heading to production, and use the Models format for serving. Projects matter more when reproducible packaging of training code is a priority. You do not need all four to get value — start with the one that solves your current problem.

Logging runs

Tracking is the entry point. You wrap your training code to log parameters, metrics, and artifacts to an MLflow server, and it records each run so you can compare them in the UI. The integration is a few lines and works with the common ML libraries, which is why Tracking is where nearly everyone begins — it delivers experiment reproducibility and comparison immediately, for little effort.

The model registry and lifecycle

Once you have models worth promoting, the Model Registry manages their lifecycle: it versions each registered model and tracks which version is in which stage — staging, production, archived. This gives you a single source of truth for "which model is in production right now," a controlled promotion path from staging to production, and the ability to roll back to a previous version. It is the piece that turns tracked experiments into managed production models.

Adopt the components you actually need

MLflow is modular by design — you are not obliged to use all of it. Start with Tracking for reproducibility, add the Model Registry when models start going to production and you need versioning and stage management, and bring in the packaging pieces when serving and reproducible runs become priorities. Adopting the whole platform before you need it is complexity you will maintain for no benefit.

MLflow's four components — Tracking, Model Registry, Models, and Projects — each address a different part of the ML lifecycle, and the way to use it well is to adopt the ones that solve your current problems: Tracking for experiment reproducibility, the Registry for versioning and promoting production models. Treat it as a modular toolkit rather than an all-or-nothing platform, and it becomes a lightweight backbone for managing models from experiment to production.

John Kihiu
Acumatica ERP Developer · Laravel Engineer

Independent software engineer in Nairobi specialising in Acumatica customisations, Laravel backends, and tax fiscalisation integrations across East and Southern Africa.