aws-ai-ml

Fine-tune, evaluate, and deploy AI models on Amazon SageMaker.

2.5k|282|Updated Apr 23, 2026
One-click install
npx skills add https://github.com/aws/agent-toolkit-for-aws --skill aws-ai-ml
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: aws-ai-ml
Source: https://github.com/aws/agent-toolkit-for-aws/tree/main/plugins/aws-core/skills/aws-ai-ml
Command: npx skills add https://github.com/aws/agent-toolkit-for-aws --skill aws-ai-ml

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires boto3, pandas, and includes scripts (resource) and references (resource) components.

What problem does it solve?

Customizing and deploying foundation models on AWS involves many decisions — choosing a base model, picking a fine-tuning technique, validating datasets, generating training code, and managing endpoints. This Skill guides AI coding agents through the entire SageMaker model customization lifecycle with validated workflows, so nothing critical gets missed.

Core Features & Use Cases

  • Model Selection & Planning: Discover models from SageMaker Hub, define use case specs, and build step-by-step customization plans (fine-tune, evaluate-first, or deploy base model).
  • Fine-Tuning & Data Preparation: Generate training code for SFT, DPO, RLVR, and RLAIF techniques; validate and transform datasets into required JSONL formats.
  • Evaluation, Deployment & Diagnostics: Evaluate models with LLM-as-Judge or Custom Scorers, deploy to SageMaker endpoints or Bedrock, manage MLflow apps, and diagnose endpoint failures with CloudWatch metrics and logs.
  • Use Case: A developer wants to fine-tune a Llama model on their customer support data. The Skill helps them select the model, validate their dataset, generate the SFT training script, evaluate the result, and deploy it to a SageMaker endpoint.

Quick Start

Ask the agent to help you fine-tune a model on SageMaker, for example: "Help me fine-tune a Qwen model on my dataset and deploy it to a SageMaker endpoint."

Frequently Asked Questions about aws-ai-ml

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

FAQPage Schema
How do I fine-tune a model on Amazon SageMaker?

Select a base model from SageMaker Hub, choose a fine-tuning technique (SFT, DPO, RLVR, or RLAIF), validate your JSONL training dataset, and generate training code from the provided templates. The workflow handles IAM roles, S3 buckets, and EULA acceptance before starting the training job.

What fine-tuning techniques does SageMaker Serverless Model Customization support?

This Skill has validated workflows for SFT (Supervised Fine-Tuning), DPO (Direct Preference Optimization), RLVR (Reinforcement Learning from Verifiable Rewards), and RLAIF. Technique availability depends on the selected model and is validated against the model's recipes.

Can I deploy a fine-tuned model to Amazon Bedrock?

Yes, both OSS models (Llama, Mistral, Qwen) and Nova models fine-tuned through SageMaker Serverless Model Customization can be deployed to SageMaker endpoints or Bedrock. JumpStart base models deploy only to SageMaker real-time endpoints.

What dataset format is required for SageMaker fine-tuning?

All training and evaluation datasets must be serialized as JSONL (one JSON object per line) and match the schema for your model type and fine-tuning technique. Non-JSONL files like parquet or CSV must be transformed first using the dataset transformation workflow.

How do I diagnose a failing SageMaker endpoint?

The endpoint diagnostics workflow collects endpoint status via DescribeEndpoint, CloudWatch metrics (invocations, errors, latency, utilization), and recent container logs. It is read-only and covers inference issues, not training job failures.

What is not covered by this SageMaker customization skill?

It does not cover Ground Truth labeling, Feature Store, HyperPod training, full fine-tuning (FFT), BYO container training, traditional ML models like XGBoost, or general-purpose AWS infrastructure. For those cases it offers best-effort guidance from general AWS knowledge.