search

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.46k stars 467 forks source link

add spmd gpu doc back #7592

Closed JackCaoG closed 3 months ago

JackCaoG commented 3 months ago

I accidentally remove GPU SPMD doc in https://github.com/pytorch/xla/pull/7283, adding it back.