Add torch_xla TPU support for ZeRO-1/2

Commit

66 days ago

Add torch_xla TPU support for ZeRO-1/2 Signed-off-by: PKUWZP <zhipeng.rainbowserie@gmail.com>