Google open-sources TorchTPU, a native PyTorch back end for TPUs

187 points - today at 4:40 PM

Source

Comments

pcmoritz today at 6:19 PM
For people interested in this: I've been working on an open-source libtpu implementation for https://tenstorrent.com/ accelerators as a side project, it can already run most of torchtpu, see https://github.com/pcmoritz/libtt (based on the tenstorrent software stack). It shows all the things involved in lowering torch to tile based accelerators.
dalemhurley today at 8:18 PM
Google and Apple are going to move consumers and ordinary business users to edge inference. This is more disruptive to the upcoming IPOs than most investors understand.
npalli today at 4:51 PM
Interesting, Amazon doing the same for AWS Trainium.

https://github.com/aws-neuron/torch-neuronx

fancyfredbot today at 6:22 PM
Cool! XLA is pretty good + potentially a better compiler infrastructure than inductor which will lead to improved performance on all hardware.

Also I guess this is the final nail in the coffin of Tensorflow.

ismailmaj today at 7:19 PM
Amazing news, I remember the days when you had to fight XLA to make PyTorch run on TPUs, it seems we finally got first-party support!

Since HW specifics live mostly in the kernel layer (Pallas or CUDA), I imagine the Jax vs PyTorch question is more of a style preference now?

nico today at 7:33 PM
Pardon my ignorance, but who else besides Google has TPUs? Are they commercially available somewhere? What’s their main application right now outside of Google?
23agt today at 5:27 PM
Google is selling TPU shovels via Cloud services and also physically to select partners. So this makes sense.

There will be an abundance of shovels. Nebius already crashed by almost half from its hype peak.

Lucasoato today at 4:48 PM
Are there TPUs outside Google Cloud?
yipinwong today at 5:07 PM
Dumb question from non-AI engineer.

Is this somehow Google's answer to today's best DeepSeek open-sourcing inferencing engine?

tuna74 today at 7:28 PM
Is the Linux driver available?
nelaggy today at 5:50 PM
so if the moat isn't the model and the moat isn't the inference layer, i guess the moat is the hardware now? what does this mean for frontier labs?
Androcles today at 5:05 PM
[flagged]
karansuthar today at 5:52 PM
[dead]