we don’t anymore have a Trainer for Tensorflow, but yeah the PyTorch Trainer supports TPU.
Or you can also use huggingface/accelerate.
1 Like
Can you give me an example on pytorch trainer?