Cache all_optimizer_states to speed up model sharding#2747

Closed

hstonec wants to merge 1 commit intopytorch:mainfrom hstonec:export-D69443708

+5-1

Commits on Feb 18, 2025

Cache all_optimizer_states to speed up model sharding (pytorch#2747 )

hstonec
authored and
facebook-github-bot
committed