Skip to content

Conversation

@kmanuwai
Copy link
Contributor

Issue #, if available:
#5513

Description of changes:
Adds a new example notebook demonstrating the complete end-to-end workflow for training and deploying models with MLflow 3.x integration on SageMaker.

What's included

  • v3-mlflow-train-inference-e2e-example.ipynb

Workflow covered

  1. Connect to SageMaker MLflow tracking server
  2. Train PyTorch model using ModelTrainer with MLflow metric/param logging
  3. Register model to MLflow Model Registry via mlflow.pytorch.log_model()
  4. Deploy from MLflow registry using ModelBuilder with MLFLOW_MODEL_PATH
  5. Test endpoint with JSON payloads
  6. Clean up resources

Prerequisites
SageMaker MLflow App (tracking server ARN required)

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant