/tpu_pytorch_xla

A step-by-step guide to setting up a TPU pod, and running your first Pytorch/XLA code on it.

Primary LanguagePython

Stargazers