Add a context manager for activation sharding.#1127

luyug wants to merge 1 commit intofacebookresearch:mainfacebookresearch/fairscale:mainfrom

luyug:act_shardluyug/fairscale:act_shardCopy head branch name to clipboard