Lightning-AI / Lightning-AI/lightning-thunder
Write custom `nn.Module`s with `DTensor`s
Open
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 1.5k
- Forks
- 121
- PR merge metrics
- No merged PRs in 30d
Description
of the equivalent semantics as SGLang's custom nn.Modules
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Use SGLang’s custom nn.Module implementations as the reference and investigate how equivalent semantics should work with DTensors. The issue does not name files, tests, or an entry point, so first locate the existing custom nn.Module and DTensor handling in the repository. Done means custom nn.Modules have equivalent semantics when written with DTensors.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- distributed-systems
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100