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