Back to the feed
devtools

Introducing TorchTPU

PyTorch Blog1 min read

TorchTPU is a new open-source PyTorch backend developed through a collaboration between Google and Meta, enabling native and efficient execution of PyTorch models on Google TPUs. It provides seamless integration, allowing developers to run models on TPUs without requiring significant code changes.

AI summary
Read at PyTorch Blog