Skip to content

Toolkit for running PyTorch training scripts on SageMaker. Dockerfiles used for building SageMaker Pytorch Containers are at https://github.com/aws/deep-learning-containers.

License

Notifications You must be signed in to change notification settings

aws/sagemaker-pytorch-training-toolkit

Folders and files

NameName
Last commit message
Last commit date

Latest commit

c0f34cf · Oct 23, 2023
Jun 24, 2020
May 4, 2020
Jun 10, 2020
Oct 20, 2023
Oct 20, 2023
Feb 19, 2020
Apr 15, 2019
Oct 23, 2023
Mar 1, 2018
Sep 20, 2022
Mar 1, 2018
Feb 6, 2020
Mar 1, 2018
Sep 20, 2022
Oct 23, 2023
Jun 10, 2020
Feb 6, 2020
Jul 8, 2022
Jul 8, 2022
Jul 8, 2022
Jul 8, 2022
Apr 16, 2018
Apr 27, 2023
Jul 8, 2022

Repository files navigation

SageMaker PyTorch Training Toolkit

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

This toolkit depends and extends the base SageMaker Training Toolkit with PyTorch specific support.

For inference, see SageMaker PyTorch Inference Toolkit.

For the Dockerfiles used for building SageMaker PyTorch Containers, see AWS Deep Learning Containers.

For information on running PyTorch jobs on Amazon SageMaker, please refer to the SageMaker Python SDK documentation.

For notebook examples: SageMaker Notebook Examples.

Contributing

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

License

SageMaker PyTorch Training Toolkit is licensed under the Apache 2.0 License. It is copyright 2018 Amazon .com, Inc. or its affiliates. All Rights Reserved. The license is available at: http://aws.amazon.com/apache2.0/