[zero] add support for `shard state dict` save/load

[Gy-Lu]

Now the state_dict of low-level zero follows the torch form.
For compatibility, the shard form(like huggingface) is needed.