sagemaker-training-toolkit

Train machine learning models within a 🐳 Docker container using 🧠 Amazon SageMaker.

APACHE-2.0 License

Downloads
153.1K
Stars
490
Committers
45
View Code on GitHub
Ecosystems: Amazon Web Services, Docker, Python

Bot releases are visible (Hide)

sagemaker-training-toolkit - v4.4.0

Published by sagemaker-bot almost 2 years ago

Features

  • integrate SMDDP collectives into smdataparallel runner
sagemaker-training-toolkit - v4.3.2

Published by sagemaker-bot almost 2 years ago

Bug Fixes and Other Changes

  • add general exception to filter
sagemaker-training-toolkit - v4.3.1

Published by sagemaker-bot almost 2 years ago

Bug Fixes and Other Changes

  • integrate upcoming dataparallel change to modelparallel
  • add unit tests for torchrun launcher and collections package deprecationWarning
sagemaker-training-toolkit - v4.3.0

Published by sagemaker-bot almost 2 years ago

Features

  • Add torch_distributed support for Trainium instances in SageMaker
sagemaker-training-toolkit - v4.2.10

Published by sagemaker-bot about 2 years ago

Bug Fixes and Other Changes

    • feature: Add neuron cores support (#21)
sagemaker-training-toolkit - v4.2.9

Published by sagemaker-bot about 2 years ago

Bug Fixes and Other Changes

  • Add SageMaker Debugger exceptions
sagemaker-training-toolkit - v4.2.8

Published by sagemaker-bot about 2 years ago

sagemaker-training-toolkit - v4.2.7

Published by sagemaker-bot about 2 years ago

Bug Fixes and Other Changes

  • improve worker node wait logic and update EFA flags
sagemaker-training-toolkit - v4.2.6

Published by sagemaker-bot about 2 years ago

Bug Fixes and Other Changes

  • Enable PT XLA distributed training on homogeneous clusters
sagemaker-training-toolkit - v4.2.5

Published by sagemaker-bot about 2 years ago

Bug Fixes and Other Changes

  • relax exception type
sagemaker-training-toolkit - v4.2.4

Published by sagemaker-bot about 2 years ago

sagemaker-training-toolkit - v4.2.3

Published by sagemaker-bot about 2 years ago

Bug Fixes and Other Changes

  • update num_processes_per_host for smdataparallel runner
sagemaker-training-toolkit - v4.2.2

Published by sagemaker-bot about 2 years ago

Bug Fixes and Other Changes

  • Removed version hardcoding for sagemaker test dependency
  • update distribution_instance_group for pytorch ddp
  • specify flake8 config explicitly
sagemaker-training-toolkit - v4.2.1

Published by sagemaker-bot about 2 years ago

Bug Fixes and Other Changes

  • handle utf-8 decoding exceptions while processing stdout and stderr streams
sagemaker-training-toolkit - v4.2.0

Published by sagemaker-bot over 2 years ago

Features

  • Heterogeneous cluster changes
sagemaker-training-toolkit - v4.1.6

Published by sagemaker-bot over 2 years ago

Bug Fixes and Other Changes

  • update: protobuf version to overlap with TF requirements
sagemaker-training-toolkit - v4.1.5

Published by sagemaker-bot over 2 years ago

Bug Fixes and Other Changes

  • Fix none exception class issue for mpi
sagemaker-training-toolkit - v4.1.4

Published by sagemaker-bot over 2 years ago

Bug Fixes and Other Changes

  • Use framework provided error class and stack trace as error message
sagemaker-training-toolkit - v4.1.3

Published by sagemaker-bot over 2 years ago

sagemaker-training-toolkit - v4.1.2

Published by sagemaker-bot over 2 years ago

Bug Fixes and Other Changes

  • fix flaky issue with incorrect rc being given
Package Rankings
Top 2.22% on Pypi.org
Badges
Extracted from project README
Latest Version Supported Python Versions Code Style: Black
Related Projects
aecc

AeCC: Autoencoders for Compressed Communication

24 Nov 2023 3
tensorflow-qnd

Quick and Dirty TensorFlow command framework to train and evaluate models and make inference

20 Dec 2016 56
generative-models

Generative Models by Stability AI

22 Jun 2023 24,227
sagemaker-python-sdk

A library for training and deploying machine learning models on Amazon SageMaker

14 Nov 2017 2,043
Made-With-ML

Learn how to design, develop, deploy and iterate on production-grade ML applications.

05 Nov 2018 36,993
ml-workspace

🛠 All-in-one web-based IDE specialized for machine learning and data science.

27 May 2019 3,406
deep-thinking

A centralized place for deep thinking code and experiments

18 Oct 2021 67
refinery-python-sdk

Official Python SDK for Kern AI refinery.

06 Jun 2021 18
DocsGPT

GPT-powered chat for documentation, chat with your documents

02 Feb 2023 14,124
mlops-course

Learn how to design, develop, deploy and iterate on production-grade ML applications.

12 Nov 2020 2,915
sagemaker-tutorial

Multi-node, multi-GPU training with PyTorch Lightning on SageMaker

12 Jan 2022 5
pytorch-template

PyTorch deep learning projects made easy.

13 Mar 2018 4,714
sagemaker-inference-toolkit

Serve machine learning models within a 🐳 Docker container using 🧠 Amazon SageMaker.

20 May 2019 385