Need help with sagemaker-tensorflow-container?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

235 Stars 139 Forks Apache License 2.0 225 Commits 6 Opened issues


Toolkit for running TensorFlow training scripts on SageMaker. Dockerfiles used for building SageMaker TensorFlow Containers are at

Services available


Need anything else?

Contributors list


SageMaker TensorFlow Training Toolkit

SageMaker TensorFlow Training Toolkit is an open-source library for using TensorFlow to train models on Amazon SageMaker.

For inference, see

SageMaker TensorFlow Inference Toolkit 

For the Dockerfiles used for building SageMaker TensorFlow Containers, see

AWS Deep Learning Containers 

For information on running TensorFlow jobs on Amazon SageMaker, please refer to the

SageMaker Python SDK documentation 

For notebook examples:

SageMaker Notebook


Please read 
__ for details on our code of conduct, and the process for submitting pull requests to us.


SageMaker TensorFlow Training Toolkit is licensed under the Apache 2.0 License. It is copyright 2018, Inc. or its affiliates. All Rights Reserved. The license is available at:

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.