Google TorchTPU: PyTorch Now Runs Natively on TPUs
Google just open-sourced TorchTPU, so PyTorch code can now run natively on Google's TPU chips, not only on GPUs. Meta's PyTorch team helped build it, and Hugging Face models move to a TPU by changing one setting. If your team is waiting on GPUs or paying a lot for them, it's worth asking whether your models could run on TPUs too. Want a deep dive? Comment DEEP DIVE. Google Cloud blog: https://cloud.google.com/blog/topics/ai-infrastructure/torchtpu-is-now-open-source-run-native-pytorch-on-tpus Hugging Face blog: https://huggingface.co/blog/hf-torchtpu Code: https://github.com/google-pytorch/torch_tpu #shorts #pytorch #googlecloud Opinions are my own.