TensorFlow scripts utilize a TPUStrategy for training on TPUs.