pytorch/xla
View on GitHub ↗ ★ 2,807 C++
Enabling PyTorch on XLA Devices (e.g. Google TPU)
13 mentions on Hacker News by 11 people: 3 stories, 10 comments · first 2019-05-08 · latest 2026-10-20
20082026
-
lol no this is a replacement for https://github.com/pytorch/xla
comment by mathisfun123 · 2026-10-20 · in Google open-sources TorchTPU, a native PyTorch back end for TPUs -
▲ 3
RFC: Evolving PyTorch/XLA for a more native experience on TPUstory by agnosticmantis · 2025-10-28 · 0 comments
-
>Cheers, > The PyTorch/XLA Team at Google Meanwhile you have an issue from 5 years ago with 0 support https://github.com/pytorch/xla/issues/202
comment by htrp · 2024-03-11 · in Who uses Google TPUs for inference in production? -
https://github.com/pytorch/xla/
comment by ypcx · 2021-10-07 · in How to train large deep learning models as a startup -
We both know (and correct me if I'm wrong) that this is an issue with the GCP cloud architecture. https://github.com/pytorch/xla/issues/1858 So you can't blame the PyTorch team. If there's anyone to blame, it's Google Cloud. In the…
comment by tmabraham · 2020-10-08 · in Grid: AI platform from the makers of PyTorch Lightning -
…has TPU support via XLA and has been used in research at the scale of billions of parameters. https://github.com/pytorch/xla
comment by Smerity · 2020-01-30 · in Standardizing OpenAI’s deep learning framework on PyTorch -
…at scale on large, distributed Cloud TPU Pod slices. Would you consider filing a GitHub issue at https://github.com/pytorch/xla or emailing pytorch-tpu@googlegroups.com to provide a bit more context about the specific issue you…
comment by zak · 2020-01-03 · in Predicting where AI is going in 2020 -
We have been working with G on TPU support for PyTorch. Have you tried it out? https://github.com/pytorch/xla
comment by jspisak · 2019-12-07 · in Tensorflow User Experience -
…are now supported for PyTorch. Here is the link some additional info including some Colab NBs: https://github.com/pytorch/xla
comment by jspisak · 2019-11-13 · in 16-inch MacBook Pro -
▲ 8
GitHub – Pytorch/Xla: Enabling PyTorch on Google TPUstory by axiomdata316 · 2019-10-14 · 0 comments
-
…Is there any way to use TPUs? PyTorch has an optional XLA device, that let's you use TPUs: https://github.com/pytorch/xla
comment by smhx · 2019-10-01 · in Tensorflow 2.0 -
▲ 2
PyTorch for TPUsstory by nl · 2019-05-08 · 0 comments
-
Well you'll be glad to know that PyTorch is available (in development form) on the TPU: https://github.com/pytorch/xla
comment by nl · 2019-05-08 · in Google’s Cloud TPU Pods are now publicly available in beta