[DRAFT] Factor out core SDPA#1561

jackzhxng wants to merge 8 commits intometa-pytorch:mainmeta-pytorch/torchtune:mainfrom

jackzhxng:jackxz/rewrite-attention-2jackzhxng/torchtune:jackxz/rewrite-attention-2Copy head branch name to clipboard