sagemaker-mlflow

Connect SageMaker environments to MLflow for tracking experiments and managing models.

69|21|Updated Jan 14, 2026
One-click install
npx skills add https://github.com/mlflow/skills --skill sagemaker-mlflow
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: sagemaker-mlflow
Source: https://github.com/mlflow/skills/tree/main/sagemaker-mlflow
Command: npx skills add https://github.com/mlflow/skills --skill sagemaker-mlflow

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires sagemaker-mlflow, and includes scripts (resource) components.

What problem does it solve?

This Skill allows users to connect their SageMaker environment with MLflow, providing a seamless way to track experiments, log metrics, and manage models directly within SageMaker.

Core Features & Use Cases

  • SageMaker MLflow Integration: Connect to SageMaker Managed MLflow as an MLflow backend.
  • Resource Discovery: Automatically discover SageMaker MLflow ARNs in the user's region.
  • Predefined Steps: Predefined commands to install the plugin, select the resource ARN, and verify the connection.
  • Use Case: Ideal for data scientists who need to track their SageMaker experiments and models within the same environment.

Quick Start

Set up MLflow integration with SageMaker by running the following command: 'connect to SageMaker MLflow'.

Frequently Asked Questions about sagemaker-mlflow

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I connect MLflow to SageMaker for experiment tracking?

To connect MLflow to SageMaker for experiment tracking, you can use predefined steps to install the necessary plugin, select your SageMaker Managed MLflow ARN, and verify the connection. This provides a seamless way to track experiments and manage models directly within SageMaker.

Do I need AWS credentials to use SageMaker Managed MLflow?

Yes, you need AWS credentials to authenticate with SageMaker when using SageMaker Managed MLflow. The integration requires valid AWS credentials alongside the sagemaker-mlflow plugin to successfully connect and manage your MLflow backend.

Can I automatically discover SageMaker MLflow ARNs in my AWS region?

Yes, the integration supports resource discovery to automatically discover SageMaker MLflow ARNs in your AWS region. This feature helps you locate and select the correct backend resource for your environment without manual lookup.

What is the best way to manage models and log metrics within a SageMaker environment?

The best way to manage models and log metrics within a SageMaker environment is by integrating SageMaker with MLflow. This allows data scientists to track experiments, log metrics, and manage models seamlessly within the same environment.

Why does my SageMaker MLflow connection require a specific plugin?

Your SageMaker MLflow connection requires the sagemaker-mlflow plugin to handle AWS credentials and enable communication between your local environment and the SageMaker Managed MLflow backend. Without it, authentication and tracking cannot be established.