Skip to main content

Open source library for creating TensorFlow containers to run on Amazon SageMaker.

Project description

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 Examples.


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:

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page