Sagemaker Python Packages

sagemaker

A library for training and deploying machine learning models on Amazon SageMaker

sagemaker-core

A library for training and deploying machine learning models on Amazon SageMaker

sagemaker-train

A library for training and deploying machine learning models on Amazon SageMaker

gluonts

Probabilistic time series modeling in Python

sagemaker-inference

Serve machine learning models within a 🐳 Docker container using 🧠 Amazon SageMaker.

sagemaker-training

Train machine learning models within a 🐳 Docker container using 🧠 Amazon SageMaker.

smdebug

Amazon SageMaker Debugger provides functionality to save tensors during training of machine learning jobs and analyze those tensors

sagemaker-pyspark

A Spark library for Amazon SageMaker.

sagemaker-studio-image-build

CLI for building Docker images in SageMaker Studio using AWS CodeBuild.

cdklabs-generative-ai-cdk-constructs

AWS Generative AI CDK Constructs are sample implementations of AWS CDK for common generative AI patterns.

sagemaker-ssh-helper

A helper library to connect into Amazon SageMaker with AWS Systems Manager and SSH (Secure Shell)

autogluon-cloud

Train and deploy ML models in the cloud

aws-analyzer

Readable reports on your AWS data from a SageMaker or Jupyter notebook: S3 buckets, DynamoDB tables, Bedrock knowledge bases and the SageMaker notebook itself. What's there, what it costs, and what to do next. Read-only, boto3 only; one file per service, or pip install aws-analyzer.

isaacus-sagemaker

A Python library for interacting with AWS SageMaker AI deployments of the Isaacus API.

modelmoat

Static analysis for the AI/ML parts of your Terraform: SageMaker networking, Bedrock and SageMaker IAM, model buckets, and vector stores.

sagemaker-tensorflow-training

Toolkit for running TensorFlow training scripts on SageMaker. Dockerfiles used for building SageMaker TensorFlow Containers are at https://github.com/aws/deep-learning-containers.

sagemaker-pytorch-training

Toolkit for running PyTorch training scripts on SageMaker. Dockerfiles used for building SageMaker Pytorch Containers are at https://github.com/aws/deep-learning-containers.

sagemaker-shim

Adapts algorithms that implement the Grand Challenge inference API for running in SageMaker

sagify

LLMs and Machine Learning done easily

kosty

Scan 30+ AWS services. Find cost waste. Detect security gaps. Map your attack surface. One command.

inference-server

Deploy your AI/ML model to Amazon SageMaker for Real-Time Inference and Batch Transform using your own Docker container image.

jina-sagemaker

Jina Embedding Models on AWS SageMaker

spotty

Training deep learning models on AWS and GCP instances

loam-geo

loam-geo — an open-source Python alternative to AWS SageMaker Geospatial (which closed to new customers in 2026). Process Sentinel-2 & Landsat imagery over STAC/COGs: spectral indices (NDVI, band-math), cloud masking, reproject/resample, temporal composites, reverse-geocoding. Execution-agnostic, runs in any region.