fsdp: improve state saving determinism and tree reconstruction#3407

coder-jayp wants to merge 1 commit intoml-explore:mainml-explore/mlx:mainfrom

coder-jayp:fsdp-saving-state-improvementcoder-jayp/mlx:fsdp-saving-state-improvementCopy head branch name to clipboard