Recover shared parameters

[ShijieZZZZ]

To address this issue:

[BUG] DeepSpeed zero_to_fp32.py script ignores some layers while creating FP32 checkpoints from DS ZeRO checkpoints.

Shared parameters that hold reference are missing when extracts fp32 weights from checkpoint. This PR recovers such shared parameters by linking them with their partners.