Register _ScaledPartial placement#2313

Aidyn-A wants to merge 3 commits intopytorch:mainpytorch/torchtitan:mainfrom

Aidyn-A:register_ScaledPartial_placementAidyn-A/torchtitan:register_ScaledPartial_placementCopy head branch name to clipboard