Skip to content

Add DCP-shardable FSDP2 optimizer state and portable resharding - #82

Closed
thad0ctor wants to merge 57 commits into
mainfrom
feat/dcp-resharding-main
Closed

Add DCP-shardable FSDP2 optimizer state and portable resharding#82
thad0ctor wants to merge 57 commits into
mainfrom
feat/dcp-resharding-main

Add DTensor optimizer contracts and sharded portable resharding

bf77e41
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar