Skip to content

Navigation Menu

Explore
By company size
By use case
By industry
View all solutions
Topics
- AI
- DevOps
- Security
- Software Development
- View all
Explore
- GitHub Sponsors
  Fund open source developers
- The ReadME Project
  GitHub community articles
Repositories
- Enterprise platform
  AI-powered developer platform
Available add-ons
Pricing

Search code, repositories, users, issues, pull requests...

Search

Clear

Search syntax tips

Provide feedback

We read every piece of feedback, and take your input very seriously.

Include my email address so I can be contacted

Saved searches

Use saved searches to filter your results more quickly

Name

Query

To see all available qualifiers, see our documentation.

You signed in with another tab or window. Reload to refresh your session. You signed out in another tab or window. Reload to refresh your session. You switched accounts on another tab or window. Reload to refresh your session.

Dismiss alert

Lightning-AI / lightning-thunder Public

Notifications You must be signed in to change notification settings
Fork 86
Star 1.3k

Code
Issues 287
Pull requests 29
Actions
Security
Insights

Additional navigation options

Code
Issues
Pull requests
Actions
Security
Insights

update FSDPTransform to delegate materialization to the MaterializationTransform#995

Merged

t-vi merged 21 commits intomainLightning-AI/lightning-thunder:mainfrom tom/fsdp-composable-step4Lightning-AI/lightning-thunder:tom/fsdp-composable-step4Copy head branch name to clipboard

Aug 20, 2024

+143-110

Conversation Commits21 (21)Checks Files changed

Commits

Commits on Aug 18, 2024

Move FSDP module transformation to transform_module
t-vi
committed
import _materialize
t-vi
committed
Update thunder/distributed/transforms/fsdp_v2.py
t-vi
authored
Update thunder/distributed/transforms/fsdp_v2.py

t-vi
and
crcrpar
authored

Commits on Aug 19, 2024

Merge remote-tracking branch 'origin/main' into tom/fsdp-composable-step1
t-vi
committed
free orig parameters after sharding
t-vi
committed
Update thunder/distributed/transforms/fsdp_v2.py
t-vi
committed
implement FSDPTransform.transform_state_dict_for_submodule
t-vi
committed
Update thunder/distributed/transforms/fsdp_v2.py

t-vi
and
crcrpar
committed
Update thunder/distributed/__init__.py
t-vi
committed
update FSDPTransform to delegate materialization to the MaterializationTransform
t-vi
committed
Apply suggestions from code review

t-vi
and
crcrpar
committed
Update thunder/distributed/transforms/fsdp_v2.py

t-vi
and
crcrpar
committed
don't use nn as a local variable name, thank you Masaki
t-vi
committed
Merge branch 'main' into tom/fsdp-composable-step3
t-vi
authored
be more thorough in copying the module before to_empty
t-vi
committed
Merge branch 'tom/fsdp-composable-step3' into tom/fsdp-composable-step4
t-vi
authored

Commits on Aug 20, 2024

better keep track of materialization
t-vi
committed
Merge branch 'main' into tom/fsdp-composable-step4
t-vi
authored
no-grad for broadcast
t-vi
committed
Apply suggestions from code review

t-vi
and
lantiga
authored

Footer

© 2025 GitHub, Inc.

Footer navigation

Terms
Privacy
Security
Status
Docs
Contact

You can’t perform that action at this time.