AWS ships an aws-ai-ml skill that lets Claude Code, Codex, and Kiro benchmark and size SageMaker endpoints

Install it with npx skills add aws/agent-toolkit-for-aws/skills/aws-ai-ml (after aws configure agent-toolkit, which needs AWS CLI 2.35+ and uv), and your agent can load-test a live endpoint, rank instance types for a model from S3, JumpStart, or Hugging Face, and diff two benchmark runs, all as SageMaker Python SDK v3 code you review and run. It runs under your own AWS credentials and asks before driving real traffic, so clean up the endpoints and S3 objects it creates to avoid charges.