DYK you can switch between TPUs and GPUs without rewriting code? Now, with native PyTorch support for TPUs, you can bring your research teams and models to TPUs as they are. For day 6 of #8daysofTPU8, learn more about TorchTPU → https:// goo.gle/3QGZvhE
Switch between TPUs and GPUs easily
You can switch between TPUs and GPUs without rewriting code, now with native PyTorch support.