MLflow
Skill Verified ActiveTrack ML experiments, manage model registry with versioning, deploy models to production, and reproduce experiments with MLflow - framework-agnostic ML lifecycle platform
To enable users to effectively track ML experiments, manage model lifecycles with versioning and registries, and deploy models to production using the MLflow framework.
Features
- Track ML experiments with parameters, metrics, and artifacts
- Manage model registry with versioning, stages, and aliases
- Deploy models to various platforms (local, cloud, serving)
- Reproduce experiments with project configurations
- Autologging for popular ML frameworks
Use Cases
- When needing to manage the complete ML lifecycle from experimentation to deployment.
- To collaborate on ML projects with versioned models and reproducible experiments.
- When deploying models to production and needing a centralized registry.
- To compare different model versions and track performance over time.
Non-Goals
- Providing a custom ML framework, instead guiding users on integrating with existing ones.
- Replacing the core ML training process, but rather enhancing its manageability and reproducibility.
- Handling the deep internals of specific model architectures, focusing on the lifecycle management aspect.
Practices
- Experiment Tracking
- Model Registry Management
- Model Deployment
- Autologging Best Practices
Prerequisites
- MLflow installed (`pip install mlflow`)
- Python environment
Execution
- info:Pinned dependenciesThe SKILL.md lists dependencies like 'mlflow, sqlalchemy, boto3' but doesn't specify pinned versions or lockfiles for the user's environment.
Maintenance
- info:Dependency ManagementThe SKILL.md lists dependencies, but there's no explicit mention of vulnerability checks or automated update mechanisms like dependabot for the user's environment.
Installation
First, add the marketplace
/plugin marketplace add Orchestra-Research/AI-Research-SKILLs/plugin install AI-Research-SKILLs@ai-research-skillsQuality Score
VerifiedTrust Signals
Similar Extensions
Mlflow
98Track ML experiments, manage model registry with versioning, deploy models to production, and reproduce experiments with MLflow - framework-agnostic ML lifecycle platform
TimesFM Forecasting
100Zero-shot time series forecasting with Google's TimesFM foundation model. Use for any univariate time series (sales, sensors, energy, vitals, weather) without training a custom model. Supports CSV/DataFrame/array inputs with point forecasts and prediction intervals. Includes a preflight system checker script to verify RAM/GPU before first use.
Build Feature Store
99Build a feature store using Feast for centralized feature management, configure offline and online stores for batch and real-time serving, define feature views with transformations, and implement point-in-time correct joins for ML pipelines. Use when managing features for multiple ML models, ensuring training-serving consistency, serving low-latency features for real-time inference, reusing feature definitions across projects, or building a feature catalog for discovery and governance.
Orchestrate Ml Pipeline
99Orchestrate end-to-end machine learning pipelines using Prefect or Airflow with DAG construction, task dependencies, retry logic, scheduling, monitoring, and integration with MLflow, DVC, and feature stores for production ML workflows. Use when automating multi-step ML workflows from data ingestion to deployment, scheduling periodic model retraining, coordinating distributed training tasks, or managing retry logic and failure recovery across pipeline stages.
Register ML Model
97Register trained models in MLflow Model Registry with version control, implement stage transitions (Staging, Production, Archived) with approval workflows, and manage model lineage with comprehensive metadata and deployment tracking. Use when promoting a trained model from experimentation to production, managing multiple model versions across development stages, implementing approval workflows for governance, rolling back to previous versions, or auditing model changes for compliance.
Weights And Biases
96Track ML experiments with automatic logging, visualize training in real-time, optimize hyperparameters with sweeps, and manage model registry with W&B - collaborative MLOps platform