sagemaker-training-toolkit and sagemaker-inference-toolkit
These are complementary tools that cover different stages of the ML pipeline: the training toolkit containerizes model development, while the inference toolkit containerizes model serving, and both are typically used together in a complete SageMaker workflow.
About sagemaker-training-toolkit
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.
About sagemaker-inference-toolkit
aws/sagemaker-inference-toolkit
Serve machine learning models within a 🐳 Docker container using 🧠 Amazon SageMaker.
Related comparisons
Scores updated daily from GitHub, PyPI, and npm data. How scores work