Join the discussion
Write your take first — we'll ask for email only when you're ready to publish.
- Hacker News
- 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?by nico
- 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.
by 23agt - Soon to be written by AI anyway, so they might as well release it now.by amelius
- 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?
by ismailmaj - 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.
by fancyfredbot - 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.by dalemhurley
- Interesting, Amazon doing the same for AWS Trainium.by npalli
- 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.by pcmoritz