Skip to content

dgallitelli/sagemaker-ai

v1.1.0MIT

Portable Agent Plugin for Amazon SageMaker AI development and operations.

commands

  • uvx awslabs.sagemaker-ai-mcp-serverlatest

env

  • FASTMCP_LOG_LEVEL

scripts

  • sagemaker-ai
  • assets/hyperpod/lifecycle-script-example.sh
  • scripts/hyperpod/check-quotas.sh
  • scripts/hyperpod/diagnose-cluster.sh
  • scripts/hyperpod/validate-prerequisites.sh
  • scripts/hyperpod/validate-vpc-config.sh
  • scripts/llm-training/fetch_dlc_images.py
  • scripts/llm-training/fetch_instance_info.py
  • scripts/llm-training/inspect_requirements.py
  • scripts/llm-training/validate_dataset.py
  • 15 more
    • templates/inference.py
    • templates/launcher.py
    • templates/llm-training/cpt_hf.py
    • templates/llm-training/dpo_trl.py
    • templates/llm-training/hyperpod/submit_slurm.sh
    • templates/llm-training/launch_training_job.py
    • templates/llm-training/lora_peft.py
    • templates/llm-training/qlora_peft.py
    • templates/llm-training/sft_trl.py
    • templates/llm-training/trainium/lora_neuron.py
    • templates/llm-training/trainium/sft_neuron.py
    • templates/train_pytorch.py
    • templates/train_qlora.py
    • templates/train_sklearn.py
    • templates/train_xgboost.py
plugin.json
name
sagemaker-ai
version
1.1.0
description
Portable Agent Plugin for Amazon SageMaker AI development and operations.
author
Davide Gallitellihttps://github.com/dgallitelli
license
MIT
mcp.json
  • awslabs.sagemaker-aistdiouvx awslabs.sagemaker-ai-mcp-server@latest

What else this package ships

These files come with the package and this site does not publish them. They are listed so you know what is there before you install it.

  • 18Python files
  • 6shell scripts
  • 3YAML files
  • cluster-config-slurm.json
  • LICENSE
View on GitHub