Google TorchTPU: PyTorch Now Runs Natively on TPUs

AI with Surya
1,371 views •

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.

Keyboard shortcuts

On. Switch them off if they collide with your assistive tools; ? still opens this sheet.

Go to

Press g then the letter.

  • gh Latest
  • gs Sources
  • gm Media
  • gv Videos
  • gp Podcasts
  • gc Calendar
  • gd Decoder
  • gz Dataviz
  • ga Datasets
  • gb Blog
  • gk Markets
  • gj Careers
  • gn Prompt Notebook

On this page

  • / Focus search, where there is one
  • t Back to top
  • ? This list
  • Esc Close