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