aws / aws/sagemaker-training-toolkit

Deepspeed Launcher

Open
#184 0 comments 1 reaction 0 assignees View on GitHub
Dominant language
Python
Stars
530
Forks
140
Avg merge
1h 12m
Merged PRs (30d)
2

Description

**Describe the feature you'd like**
As LLM fine tuning demand is growing, we see many customers using a very similar framework( Hugginface, Pytorch, Deepspeed) due to added benefits and scale that comes with deepspeed.
**How would this feature be used? Please describe.**
Today customers have to write too much of a boiler plate code to get this working in SageMaker, shh-ing between container, waiting for the setup to make the connection between hosts and be able to then use deepspeed library for training. We belive having this OOTB from sagemaker-training-toolkit will be super useful to speed up the experimentation and model training cycles of scientists

**Describe alternatives you've considered**
Alternative is to build your bespoke scripts and solutions.

**Additional context**
I am an internal customer and currently know atleast 4 different teams ( and growing) working on very similar model.

Contributor guide

Open the contributing guide

Research direction

No files, tests, or entry points are named; start by mapping how sagemaker-training-toolkit currently launches distributed training and reviewing the referenced Hugging Face, PyTorch, and DeepSpeed workflow. Done means an out-of-the-box DeepSpeed launcher removes the SSH, host-connection, and setup boilerplate described here and is usable by the affected training teams.

Written by the indexing model from the issue text.

Assessment

Tech stack
aws, python, pytorch
Domain
cloud, machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.