Skip to the content
Agent Tools

sagemaker-ai

Partner, reviewed

Build, train, and deploy AI models with deep AWS AI/ML expertise brought directly into your coding assistants, covering the surface area of Amazon SageMaker AI.

Plugin · by awslabs · runs in Claude Code, Cowork, claude.ai · 19 skills, 1 MCP server · listed 2026-04-17

Claude Code

claude plugin install sagemaker-ai

Skills (19)

  • Skill dataset-evaluation

    Validates dataset formatting and quality for SageMaker model fine-tuning (SFT, DPO, or RLVR). Use when the user says "is my dataset okay", "evaluate my data", "check my training data", "I have my…

    Any agent

    npx skills add awslabs/agent-plugins --skill dataset-evaluation
  • Skill dataset-transformation

    Generates code that transforms datasets between ML schemas for model training or evaluation. Use when the user says "transform", "convert", "reformat", "change the format", or when a dataset's…

    Any agent

    npx skills add awslabs/agent-plugins --skill dataset-transformation
  • Skill directory-management

    Manages project directory setup and artifact organization. Use when starting a new project, resuming an existing one, or when a PLAN.md needs to be associated with a project directory.

    Any agent

    npx skills add awslabs/agent-plugins --skill directory-management
  • Skill finetuning

    Generates code that fine-tunes a base model using SageMaker serverless training jobs. Use when the user says "start training", "fine-tune my model", "I'm ready to train", or when the plan reaches…

    Any agent

    npx skills add awslabs/agent-plugins --skill finetuning
  • Skill finetuning-technique

    Selects a fine-tuning technique (SFT, DPO, RLVR, or RLAIF) for the user's use case and validates it against the selected model's available recipes.

    Any agent

    npx skills add awslabs/agent-plugins --skill finetuning-technique
  • Skill hyperpod-cluster-debugger

    Diagnose and remediate cluster-wide HyperPod (EKS or Slurm) problems — creation / deployment failures (CloudFormation, EFA health check, lifecycle scripts, capacity), EKS access, node replacement…

    Any agent

    npx skills add awslabs/agent-plugins --skill hyperpod-cluster-debugger
  • Skill hyperpod-issue-report

    Generate comprehensive issue reports from HyperPod clusters (EKS and Slurm) by collecting diagnostic logs and configurations for troubleshooting and AWS Support cases.

    Any agent

    npx skills add awslabs/agent-plugins --skill hyperpod-issue-report
  • Skill hyperpod-nccl

    Diagnose NCCL failures and adjacent training-pod failures on HyperPod GPU clusters (EKS or Slurm) — training hangs, AllReduce / collective-op timeouts, EFA or libfabric errors, rendezvous failures…

    Any agent

    npx skills add awslabs/agent-plugins --skill hyperpod-nccl
  • Skill hyperpod-node-debugger

    Diagnose and remediate per-node issues on a HyperPod cluster (EKS or Slurm) — a specific node is unhealthy, unresponsive, stuck, or needs replacing.

    Any agent

    npx skills add awslabs/agent-plugins --skill hyperpod-node-debugger
  • Skill hyperpod-performance-debugger

    Diagnose performance issues on Amazon SageMaker HyperPod clusters — uneven NCCL bandwidth across nodes and poor filesystem throughput. Read-only.

    Any agent

    npx skills add awslabs/agent-plugins --skill hyperpod-performance-debugger
  • Skill hyperpod-slurm-debugger

    Diagnostic-only skill for Slurm scheduler and node-daemon issues on Amazon SageMaker HyperPod Slurm clusters. Scope mirrors the HyperPod troubleshooting guide.

    Any agent

    npx skills add awslabs/agent-plugins --skill hyperpod-slurm-debugger
  • Skill hyperpod-ssm

    Remote command execution and file transfer on SageMaker HyperPod cluster nodes via AWS Systems Manager (SSM). This is the primary interface for accessing HyperPod nodes — direct SSH is not available.

    Any agent

    npx skills add awslabs/agent-plugins --skill hyperpod-ssm
  • Skill hyperpod-version-checker

    Check and compare software component versions on SageMaker HyperPod cluster nodes - NVIDIA drivers, CUDA toolkit, cuDNN, NCCL, EFA, AWS OFI NCCL, GDRCopy, MPI, Neuron SDK (Trainium/Inferentia)…

    Any agent

    npx skills add awslabs/agent-plugins --skill hyperpod-version-checker
  • Skill model-deployment

    Generates code that deploys fine-tuned models from SageMaker Serverless Model Customization to SageMaker endpoints or Bedrock.

    Any agent

    npx skills add awslabs/agent-plugins --skill model-deployment
  • Skill model-evaluation

    Generates python code that evaluates SageMaker models. Supports two evaluation types: LLM-as-Judge and Custom Scorer. Use when the user says "evaluate my model", "run a benchmark", "test model…

    Any agent

    npx skills add awslabs/agent-plugins --skill model-evaluation
  • Skill model-selection

    Selects a base model for the user's use case by querying SageMaker Hub. Use when the user asks which model to use, wants to select or change their base model, mentions a model name or family (e.g.…

    Any agent

    npx skills add awslabs/agent-plugins --skill model-selection
  • Skill planning

    Discovers user intent and generates a structured, step-by-step plan for model customization workflows. This skill must always be activated alongside any other skill when the user's request relates…

    Any agent

    npx skills add awslabs/agent-plugins --skill planning
  • Skill sdk-getting-started

    Validates the user's environment for SageMaker AI operations — checks SDK version, AWS region, and execution role. Use when the user says "set up", "getting started", "check my environment"…

    Any agent

    npx skills add awslabs/agent-plugins --skill sdk-getting-started
  • Skill use-case-specification

    Creates a reusable use case specification file that defines the business problem, stakeholders, and measurable success criteria for model customization, as recommended by the AWS Responsible AI Lens.

    Any agent

    npx skills add awslabs/agent-plugins --skill use-case-specification

From Anthropic's public plugin directory, read on 2026-10-16. Find a skill by task.