README.md

SageMaker Project Templates

This module creates organizational SageMaker Project Templates.

The templates are registered in Service Catalog and available via SageMaker Studio Classic.

Architecture

Project Templates

The module contains ogranizational SageMaker Project Templates vended as Service Catalog Products. Using the templates is available through SageMaker Studio Classic and AWS Service Catalog.

Train a model on Abalone dataset with XGBoost Template

The template contains an example SageMaker Pipeline to train a model on Abalone dataset using XGBoost, and perform model evaluation.

The template is based on basic multi-account template from AWS Enterprise MLOps Framework.

Batch Inference Template

This project template contains SageMaker pipeline that performs batch inference.

Huggingface Model Import Template

This project template contains SageMaker pipeline that imports a hugging face model based on model id and access token inputs.

Multi-account Model Deployment Template

The template contains an example CI/CD pipeline to deploy the model endpoints to multiple AWS accounts.

The template is based on basic multi-account template from AWS Enterprise MLOps Framework.

Inputs and outputs:

Required inputs:

• portfolio-access-role-arn - the ARN of the IAM Role used to access the Service Catalog Portfolio or SageMaker projects

Optional Inputs:

• portfolio-name - name of the Service Catalog Portfolio
• portfolio-owner - owner of the Service Catalog Portfolio
• dev-vpc-id - id of VPC in dev environment
• dev-subnet-ids - list of subnet ids
• dev-security-group-ids - list of security group ids
• pre-prod-account-id - pre-prod account id
• pre-prod-region - pre-prod region
• pre-prod-vpc-id - id of VPC in pre-prod environment
• pre-prod-subnet-ids - list of subnet ids
• pre-prod-security-group-ids - list of security group ids
• prod-account-id - prod account id
• prod-region - prod region
• prod-vpc-id - id of VPC in prod environment
• prod-subnet-ids - list of subnet ids
• prod-security-group-ids - list of security group ids

Sample manifest declaration

name: templates
path: modules/sagemaker/sagemaker-templates
targetAccount: primary
parameters:
  - name: portfolio-access-role-arn
    valueFrom:
      moduleMetadata:
        group: sagemaker-studio
        name: studio
        key: LeadDataScientistRoleArn
  # Below are the optional properties passed to the multi-account CI/CD deployment template
  - name: dev-account-id
    valueFrom:
      envVariable: PRIMARY_ACCOUNT
  - name: dev-region
    valueFrom:
      envVariable: PRIMARY_REGION
  - name: dev-vpc-id
    valueFrom:
      moduleMetadata:
        group: networking
        name: networking
        key: VpcId
  - name: dev-subnet-ids
    valueFrom:
      moduleMetadata:
        group: networking
        name: networking
        key: PrivateSubnetIds
  - name: pre-prod-account-id
    valueFrom:
      envVariable: PRE_PROD_ACCOUNT
  - name: pre-prod-region
    valueFrom:
      envVariable: PRE_PROD_REGION
  - name: pre-prod-vpc-id
    valueFrom:
      moduleMetadata:
        group: networking
        name: networking-pre-prod
        key: VpcId
  - name: pre-prod-subnet-ids
    valueFrom:
      moduleMetadata:
        group: networking
        name: networking-pre-prod
        key: PrivateSubnetIds
  - name: prod-account-id
    valueFrom:
      envVariable: PROD_ACCOUNT
  - name: prod-region
    valueFrom:
      envVariable: PROD_REGION
  - name: prod-vpc-id
    valueFrom:
      moduleMetadata:
        group: networking
        name: networking-prod
        key: VpcId
  - name: prod-subnet-ids
    valueFrom:
      moduleMetadata:
        group: networking
        name: networking-prod
        key: PrivateSubnetIds

Outputs (module metadata):

• ServiceCatalogPortfolioName - the name of the Service Catalog Portfolio
• ServiceCatalogPortfolioOwner - the owner of the Service Catalog Portfolio

Example Output:

{
  "ServiceCatalogPortfolioName": "MLOps SageMaker Project Templates",
  "ServiceCatalogPortfolioOwner": "administrator"
}