In this blog post, we introduce concepts to generate and analyze traces to debug PyTorch training performance on TPU VM.

Source