Write For Us

Profiling Pytorch/XLA on TPUs with XProf

E-Commerce Solutions SEO Solutions Marketing Solutions
12 Views

  • Thanks! Share it with your friends!

    You disliked this video. Thanks for the feedback!

ava
Published
Unlock the full potential of your PyTorch models running on Google TPUs. We'll look at how to profile PyTorch/XLA workloads on TPUs using XProf, and learn how to identify and eliminate bottlenecks in your training pipeline, ensuring you're getting maximum performance from the TPU hardware.

Resources:
PyTorch/XLA GitHub → https://goo.gle/4pFTA8p
XProf Documentation → https://goo.gle/3Y5v7gU

Subscribe to Google for Developers → https://goo.gle/developers

Speaker: Chris Achard
Products Mentioned: Google AI
Category
Project
Tags
Google, developers, Purpose: Learn;
Sign in or sign up to post comments.
Be the first to comment

Up Next