Lightning-AI / Lightning-AI/lightning-thunder

Question about Lightning Thunder and DeepSpeed

Open
#1,904 2 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

documentation
Dominant language
Python
Stars
1.5k
Forks
121
PR merge metrics
No merged PRs in 30d

Description

## Question

Great persentation at GTC.
Does Lightning Thunder work with DeepSpeed out of the box (assuming I didn't use FSDP) but used all other fusing, compilation options? Also how does FSDP interact with DeepSpeed?

* by DeepSpeed I specifically mean ZeRO 1, 2, 3, 2 + offload, and, 3 + offload

cc @borda @lantiga

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

The issue names no files or tests. Start by reviewing Lightning Thunder's integration points and the DeepSpeed ZeRO/FSDP documentation, then verify each listed ZeRO and offload combination; done means a documented compatibility answer covering the requested combinations.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
distributed-systems, machine-learning
Issue type
Documentation
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.