by aws

Toolkit for running TensorFlow training scripts on SageMaker. Dockerfiles used for building SageMake...

224 Stars 134 Forks Last release: 2 months ago (v20.1.2) Apache License 2.0 216 Commits 65 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:


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.