Mask the full weight for shared_across_batch#323

danbraunai-goodfire wants to merge 1 commit intomaingoodfire-ai/spd:mainfrom

feature/pgd-weight-maskgoodfire-ai/spd:feature/pgd-weight-maskCopy head branch name to clipboard