pytorch

510339c0 - [FSDP][2/N] Refactor state dict hook registration (#90777)

[FSDP][2/N] Refactor state dict hook registration (#90777) This PR includes some follow-ups from the previous PR to clean up the code. Pull Request resolved: https://github.com/pytorch/pytorch/pull/90777 Approved by: https://github.com/rohan-varma