Model Selection
AWS LabsOfficialPublished by AWS Labs
- Skill
- Research and analysis
- Data and AI
- Apache-2.0
- Low risk
- Has scripts
AutoGineer summary
Written by AutoGineer from the skill’s own files, not by the author
The model-selection skill streamlines the process of choosing a base model for AI development. It interacts with the user to identify their requirements, queries available SageMaker Hubs to list compatible models, and provides transparency regarding model licenses and performance benchmarks. By ensuring the exact Hub model ID is resolved, it prepares the user for subsequent development or fine-tuning tasks.
What it does
- Verifies the active AWS region for SageMaker access.
- Discovers and lists available SageMaker Hubs for model retrieval.
- Retrieves and presents a list of available models with their associated licenses.
- Cross-references model options with provided benchmark data.
- Confirms the final model selection with the user to ensure alignment with their goals.
When to use it
- When you need to select or change your base model for a project.
- When you are unsure which model best fits your specific use case.
- When you need to resolve a specific Hub model ID for a known model family.
- When you want to evaluate base models before proceeding to fine-tuning.
Advantages
- Ensures model selection is based on available resources in the user's specific AWS region.
- Provides clear visibility into model licensing requirements.
- Integrates benchmark data to support informed decision-making.
- Standardizes the model selection process to prevent configuration errors.
Usage examples
- “I need to pick a base model for my new project, what are my options?”
- The agent will check your AWS region, query the available SageMaker Hubs, and present a list of models with their licenses and benchmark context to help you decide.
- “I want to use a Llama model, can you help me set that up?”
- The agent will identify the available Llama models in your Hub, display their specific license details, and ask you to confirm the exact model ID to proceed.
Install
Download the skill as a ZIP:
https://autogineer.com/agent-skills/awslabs/model-selection/downloadUnzip it into the cross-client skills folder of your project:
mkdir -p .agents/skills && unzip model-selection.zip -d .agents/skills
Documentation by AWS Labs
Model Selection
Guides the user through selecting a base model based on their use case.
When to Use
User asks which model to use
User wants to select or change their base model
User mentions a model name or family (e.g., "Llama", "Mistral", "Nova") — the exact Hub model ID still needs to be resolved
User wants to evaluate a base model before deciding whether to finetune
Prerequisites
A
use_case_spec.mdfile exists. If not, activate the use-case-specification skill to generate it first.
Workflow
Step 1: Check Region
Run:
python -c "import boto3; print(boto3.session.Session().region_name)"None→ STOP. Tell user: "Set your region viaexport AWS_DEFAULT_REGION=us-west-2oraws configure."Set → store REGION in context, continue.
Step 2: Discover Hub
List all available SageMaker Hubs in the user's region by calling the SageMaker
ListHubsAPI using theaws___call_awstool.From the results, filter out any hub whose
HubDescriptioncontains "AI Registry" — these do not contain JumpStart models.The remaining hubs are eligible (e.g.,
SageMakerPublicHuband any private hubs).If exactly one eligible hub exists, use it automatically — do not ask the user.
If multiple eligible hubs exist, present them to the user and ask which one to use. Example:
I found the following model hubs: - SageMakerPublicHub — SageMaker Public Hub - Private-Hub-XYZ — Private Hub models Which hub would you like to use?Store the selected hub name for use in subsequent steps.
Step 3: Select Base Model
First, retrieve all available SageMaker Hub model names by running: python model-selection/scripts/get_model_names.py <hub-name>.
Present all available models to the user with their licenses before making any recommendations. Cross-reference the model list with references/model-licenses.md and display each as <model name> - [<license>](<url>). For example: "Qwen3-4B - Apache 2.0"
If you already know the model the user wants to use (from conversation context or planning files), confirm that it's in the list, display its license, and move on. Otherwise, help the user pick a model following the instructions in references/model-selection.md.
Important: Make sure to remember this list of available models when helping with model selection. Don't recommend a model that's not available to the user.
Step 4: Confirm Selection
Present a summary to the user:
Here's what we've selected:
- Base model: [model name]Ask if they'd like to proceed with this model.
References
references/model-selection.md— Model selection instructions and benchmark descriptionsreferences/model-licenses.md— Model license information for display during model selection
Files
- ATTRIBUTION.mdAdded by AutoGineer914 B
- LICENSELicence file added by AutoGineer9.9 KiB
- NOTICELicence file added by AutoGineer89 B
- SKILL.md3.3 KiB
- references/benchmarks/agenticIndex.md2.8 KiB
- references/benchmarks/codingIndex.md2.8 KiB
- references/benchmarks/gpqa.md2.8 KiB
- references/benchmarks/hle.md2.8 KiB
- references/benchmarks/ifbench.md2.7 KiB
- references/benchmarks/intelligenceIndex.md2.7 KiB
- references/benchmarks/mmmuPro.md3.0 KiB
- references/benchmarks/tau2.md2.8 KiB
- references/model-licenses.md7.8 KiB
- references/model-selection.md8.0 KiB
- scripts/get_model_names.pyScript1.0 KiB
references/model-selection.md
Model Selection
Help the user choose a base model for finetuning. This reference applies to both Nova and OSS model paths — benchmark data helps compare models regardless of family.
Select the most relevant benchmark(s), then provide the user with a list of available models listed in performance order for that benchmark.
Understanding the use case
Use what you already know from the conversation — the user may have described their task, domain, data, or goals. Make sure you're familiar with use_case_spec.md if it exists. Be sure to think about the success criteria from the user, if any has been documented. If you still don't have enough to map to relevant benchmarks, ask the user to describe their use case and success criteria in more detail.
Making a recommendation
Selecting relevant benchmark(s)
Read the benchmark descriptions below and identify which 2-3 benchmarks are most relevant to the user's task.
Understand the usecase: Model Selection is an essential step in model customization. You MUST make sure you understand the customer's use case well enough to select the proper benchmark. If you do not understand the usecase in enough detail, you MUST ask follow-up questions.
Criteria for Benchmark Selection: Picking the most relevant benchmarks is essential, because the user bases their model selection on this. You must think carefully about which benchmarks are really the most fitting. Frequently, intelligence index is the best, so make sure there is a good reason to rank anything else above that. Think about which benchmarks are testing model functionality that is the most relevant to what the user is trying to do.
Data Source: Artificial Analysis (artificialanalysis.ai). Independent evaluator, consistent methodology across all models. Data extracted May 2026.
Intelligence Index
Artificial Analysis's composite overall quality score, combining multiple benchmarks into a single ranking. It's an index on AA's own scale where higher is better.
Consider this: As a default ranking, when no other benchmark seems to fit. This MUST always be in the top 2-3 recommended benchmarks, because it's relevant to all tasks.
GPQA
The hardest 198 questions from the GPQA benchmark — graduate-level multiple-choice in biology, physics, and chemistry, written by PhD-level domain experts. Questions are "Google-proof" — skilled non-experts with unrestricted web access score only 34%, so the benchmark tests genuine scientific reasoning rather than information retrieval.
Consider this: For tasks related to scientific reasoning, particularly in the fields of biology, physics, and chemistry. Could be a good proxy for general scientific knowledge or the ability to think critically or logically.
IF-Bench
Tests precise control over text output — the model must satisfy unusual, mechanically verifiable constraints like placing a specific word at an exact position in a sentence, using exactly N numbers, or ensuring no two consecutive words share the same first letter. 58 novel constraint types designed to be harder than standard instruction-following tests.
Consider this: For any task requiring tight control over model output, where the user has very specific requirements about features that must be present in the model output. Could be a good proxy for instruction following, although the benchmark is really about precise instruction following in unusually difficult contexts
MMMU-Pro
College-level multimodal questions requiring both vision and reasoning across six disciplines: Art & Design, Business, Science, Health & Medicine, Humanities & Social Science, and Tech & Engineering. Image types include charts, diagrams, tables, maps, and chemical structures. The "Pro" version filters out questions answerable without the image, so it specifically tests integrated visual-textual understanding.
Consider this: For any task combining visual and textual inputs — chart/diagram interpretation, visual reasoning, document understanding. Only relevant if the user's finetuning task involves images.
τ²-bench
Simulates multi-turn customer service conversations where both the agent and a simulated user actively modify a shared state (e.g., databases, account records). The agent must follow domain-specific policies, use API tools, and resolve customer requests. Scores are from the telecom domain.
Consider this: For situations where accurate tool use is important, for multi-turn conversation, and situations where instruction following or adherence to business logic is important. Be careful not to over-index on this for customer service conversations.
HLE (Humanity's Last Exam)
2,500 expert-written questions across dozens of subjects including mathematics, humanities, and natural sciences, designed to be at the frontier of human knowledge. Questions can't be answered via internet retrieval. Scores are very low — most models cluster in the single digits.
Consider this: For tasks where reasoning, the ability to think logically, and the ability to draw connections between different concepts to form an analysis or conclusion is important. This benchmark may not show a large difference between most models.
Coding Index
Artificial Analysis's composite coding score, combining Terminal-Bench Hard (agentic software engineering, system administration, and data processing in terminal environments) and SciCode (scientist-curated coding problems across 16 scientific disciplines).
Consider this: For software engineering tasks, scientific/research computing, system administration automation, any task where the model needs to write and/or execute code .
Agentic Index
Artificial Analysis's composite agentic score, combining GDPval-AA (real-world tasks across 44 occupations and 9 industries, with shell access and web browsing in an agentic loop) and τ²-bench Telecom.
Consider this: For autonomous agents, workflow automation, tool-using assistants, any use case where the model needs to independently use tools, browse the web, or carry out multi-step tasks.
Presenting relevant benchmarks
After picking 2-3 relevant benchmarks from the above list (including Intelligence Index), present them in ranked order to the user, with a short explanation of what the benchmark is, and why you think it's relevant:
"In order to select a model, it's helpful to look at performance on public benchmarks. After considering several benchmarks, I think these are the most relevant to your task: " "1. [benchmark]: [short description of benchmark]. [Why this is relevant] "2. [benchmark]: [short description of benchmark]. [Why this is relevant]
(Include a 3rd only if you think there are 3 relevant benchmarks)
"Which of these do you think are relevant? After you pick, I'll show you the model rankings."
'' wait for user
Presenting Models for Selection
After the user tells you which benchmarks they think are relevant, you need to present the models from the corresponding benchmark file in /benchmarks. Present the table exactly as it appears in the benchmark table, skipping any rows for models that are not in the user's list of available models. Double check your work to avoid hallucinations:
"[Benchmark 1:] "[table from benchmark file]
"========================="
"[Benchmark 2:] "[table from benchmark file]
Give a 1-2 sentence analysis of what these benchmarks tell us. Keep in mind that most users balance performance and cost requirements.
Ask the user to select a model:
"Given this information, which model would you like to select?"
''wait for user
If the user has any questions about model selection, answer them to the best of your ability, leaning on the benchmarks as much as possible and being transparent about your knowledge gaps and confidence.
Once the user has chosen a model, consider that the chosen base model. This workflow is complete.
Requirements and compatibility
Works with: Claude Code, Claude apps, OpenAI Codex, GitHub Copilot, Cursor, Gemini CLI, Kiro, OpenCode, Goose, Amp, Roo Code, JetBrains Junie, Factory Droid, Devin (ex-Windsurf), Mistral Vibe, OpenHands, Letta
“Works with” means the tool reads the open Agent Skills format; a skill written for one tool may use features others lack.
Needs: Needs Python
Security
What our scanner noticed· Low risk
- Scripts that your agent may run
This skill includes scripts that your agent may run. Read them before you use it.
A static scan is not a guarantee. Skills are third-party content: review a skill before you let an agent use it.
Attribution and licence
- Published by
- AWS Labs
- Publisher
- AWS Labs (Official)
- Source
- awslabs/agent-plugins · plugins/sagemaker-ai/skills/model-selection · commit fbefb37
- Licence
- Apache License 2.0 (redistribution allowed)
- LICENSE (added by AutoGineer from LICENSE)
- NOTICE (added by AutoGineer from NOTICE)
- ATTRIBUTION.md (written by AutoGineer)
- Synced
AutoGineer is not affiliated with or endorsed by AWS Labs or AWS Labs.
Download ZIP
The ZIP holds the skill’s files, its licence files and an ATTRIBUTION.md. Download model-selection.zip