add shard_predicates to fsdp2#1014

JKSenthil wants to merge 1 commit intometa-pytorch:mastermeta-pytorch/tnt:masterfrom

JKSenthil:export-D77236696JKSenthil/tnt:export-D77236696Copy head branch name to clipboard