The documentation page PERF_TRAIN_TPU doesn’t exist in v4.36.1, but exists on the main version. Click here to redirect to the main version of the documentation.