Cloud TPU PyTorch/XLA user guide | Google Cloud
cloud.google.com › tpu › docsFeb 01, 2022 · Run ML Workloads With PyTorch/XLA. Before starting the procedures in this guide, set up a TPU VM and ssh into it as described in Prepare a Google Cloud Project. Key Point: Throughout this guide, a prefix of (vm) $ means you should run the command on the TPU VM instance. The $ alone prompt means you should run this in your local shell.