[FSDP] Optim state chkpt: key by param name, not ID (#74879)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/74879
Test Plan: Imported from OSS
Reviewed By: zhaojuanmao
Differential Revision: D35203958
Pulled By: awgu
fbshipit-source-id: 449edc7b705691941ddb021ae8f98c416b97084d
(cherry picked from commit 07bd75995d3648028e50b0c0dafceeed3cded074)
