aws/sagemaker-training-toolkit
Train machine learning models within a 🐳 Docker container using 🧠 Amazon SageMaker.
This project helps machine learning engineers or data scientists train their custom machine learning models within isolated Docker containers using Amazon SageMaker. You provide your training script and dependencies inside a Docker image, and the toolkit handles the environment setup, allowing SageMaker to run your training code efficiently. The output is a trained model ready for deployment.
535 stars.
Use this if you are an MLOps engineer or data scientist who needs to train machine learning models on AWS SageMaker using custom Docker containers for specialized environments or dependencies.
Not ideal if you prefer to use SageMaker's pre-built images and functionalities without needing custom containerization for your training process.
Stars
535
Forks
139
Language
Python
License
Apache-2.0
Category
Last pushed
Jan 16, 2026
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/aws/sagemaker-training-toolkit"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Compare
Related frameworks
aws/sagemaker-python-sdk
A library for training and deploying machine learning models on Amazon SageMaker
aws/amazon-sagemaker-examples
Example 📓 Jupyter notebooks that demonstrate how to build, train, and deploy machine learning...
aws/sagemaker-xgboost-container
This is the Docker container based on open source framework XGBoost...
aws-samples/aws-ml-enablement-workshop
組織横断的にチームを組成し、機械学習による成長サイクルを実現する計画を立てるワークショップ
aws-deepracer-community/deepracer-on-the-spot
Repo for running DeepRacer on Spot or Standard instances to save money versus the AWS DeepRacer console