Decouple FSDP all-gather tagging from activation checkpointing#336

fmassa wants to merge 1 commit intomainmeta-pytorch/autoparallel:mainfrom

fmassa/ac_fsdp_taggingmeta-pytorch/autoparallel:fmassa/ac_fsdp_taggingCopy head branch name to clipboard