Home

Welcome to ML Container Creator

Simplify your machine learning deployments on AWS SageMaker

Quick Start

Generate production-ready SageMaker containers in minutes:

npm install -g yo generator-ml-container-creator
yo ml-container-creator

Documentation

Roadmap & Feature Requests - See what's coming next and vote on features

Full Documentation: https://awslabs.github.io/ml-container-creator (Under Construction)

API Reference - Generator options and configuration (Under Construction)

Examples - Sample projects and use cases (Under Construction)

FAQ - Common questions and troubleshooting (Under Construction)

Supported Frameworks

Framework | Model Server | Instance Type | Status (Pending) scikit-learn | Flask/FastAPI | CPU/GPU | Stable (Pending) XGBoost | Flask/FastAPI | CPU/GPU | Stable (Pending) TensorFlow | Flask/FastAPI | CPU/GPU | Stable (Pending) Transformers | vLLM/SGLang | GPU | Beta (Pending)

Contributing

Contributing Guide - How to contribute to the project

Support

Report Issues: https://github.com/awslabs/ml-container-creator/issues

Community Discussions: https://github.com/awslabs/ml-container-creator/discussions

Made with love by the ML community, for the ML community