News for hackers. Stuff that matters.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by dalemhurley on Tuesday October 20, 2026 @04:18PM (#50172063)
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.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by anuptalwalkar on Tuesday October 20, 2026 @07:24PM (#50174078)
I would love to see them move off into edge computing. There should be some autonomy and siloed experience in day to day usage.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by tjwebbnorfolk on Tuesday October 20, 2026 @06:50PM (#50173780)
I would love to be able to buy a TPU for inference
Re:Google open-sources TorchTPU, a native PyTorch back end f... by gnabgib on Tuesday October 20, 2026 @09:06PM (#50174830)
Re:Google open-sources TorchTPU, a native PyTorch back end f... by Dr4kn on Wednesday October 21, 2026 @02:19AM (#50176742)
They are just wildly outdated for most tasks. Even projects like frigate suggest buying different hardware. Coral was released in 2019 and hasn't been updated once. With a jetson nano or even a raspi ai hat you get way more tops
Re:Google open-sources TorchTPU, a native PyTorch back end f... by tjwebbnorfolk on Wednesday October 21, 2026 @11:24AM (#50182602)
Ok I'll clarify: I want to buy a current-ish generation of TPU that can run LLMs. Coral is like 8 years old now
Re:Google open-sources TorchTPU, a native PyTorch back end f... by n8henrie on Thursday October 22, 2026 @09:49AM (#50195317)
I've been really hoping to get Whisper running on a coral, but it can't even do that. There was recently some progress getting it to export to tflite (and even an official example added) but the coral is missing some instructions.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by emmelaich on Tuesday October 20, 2026 @09:07PM (#50174839)
You can buy a Google Coral TPU, does that count?
Re:Google open-sources TorchTPU, a native PyTorch back end f... by Dr4kn on Wednesday October 21, 2026 @02:31AM (#50176835)
With them being released in 2019 with zero hardware revisions they are just outdated for almost every workload
Re:Google open-sources TorchTPU, a native PyTorch back end f... by lofties on Tuesday October 20, 2026 @09:03PM (#50174811)
Google, Apple, and don't forget our friends DeepSeek!
Re:Google open-sources TorchTPU, a native PyTorch back end f... by villgax on Wednesday October 21, 2026 @09:36AM (#50180733)
Yeah right, try generating an image on Apple Silicon or an Android phone with any of the Qwen or Flux family of models, let alone videos or prefill heavy LLMs.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by pcmoritz on Tuesday October 20, 2026 @02:19PM (#50170447)
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.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by asparagui on Tuesday October 20, 2026 @03:03PM (#50171090)
Very cool! I have been using Lean to drive PJRT like so: https://github.com/brettkoonce/lean4-mlir and in theory your code could unlock another platform!
Re:Google open-sources TorchTPU, a native PyTorch back end f... by pcmoritz on Tuesday October 20, 2026 @03:07PM (#50171157)
Thanks for sharing, this is super cool! Right now my approach to correctness it to just run as many tests as possible and fix them (torchtpu greatly expands the amount of available tests), but formal verification is also super cool. Having a look at your book https://lean.brettkoonce.com/blueprint.pdf

The Tenstorrent ISA is documented btw https://github.com/tenstorrent/tt-isa-documentation, I wonder if one could do the verification end-to-end with the hardware.

Re:Google open-sources TorchTPU, a native PyTorch back end f... by yhjc2692 on Tuesday October 20, 2026 @03:00PM (#50171042)
This is super cool!
Re:Google open-sources TorchTPU, a native PyTorch back end f... by pcmoritz on Tuesday October 20, 2026 @03:04PM (#50171102)
Thanks for the encouragement, if you have the hardware and end up trying it out I'd love to get your feedback (feel free to drop an issue in the repo) :)
Re:Google open-sources TorchTPU, a native PyTorch back end f... by yhjc2692 on Tuesday October 20, 2026 @03:07PM (#50171163)
I would love to do that, but unfortunately don’t have one on hand (yet?). In fact, I’m curious - did you buy a TT device to work on this, or are you using a cloud provider like Koyeb?
Re:Google open-sources TorchTPU, a native PyTorch back end f... by pcmoritz on Tuesday October 20, 2026 @03:11PM (#50171212)
I first built my own to try it out (the exact hardware is described in https://relativity.cc/hardware/ under "one Blackhole card") and when I liked the hardware I got a quietbox 2. It is very high quality but a little expensive (but inexpensive compared to other LLM hardware and it has excellent interconnects which is very rare in the consumer/dev space).
Re:Google open-sources TorchTPU, a native PyTorch back end f... by npalli on Tuesday October 20, 2026 @12:51PM (#50169215)
Interesting, Amazon doing the same for AWS Trainium.

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

Re:Google open-sources TorchTPU, a native PyTorch back end f... by AnotherGoodName on Tuesday October 20, 2026 @02:48PM (#50170870)
Pytorch conference is happening today for explaining the coincidental timing
Re:Google open-sources TorchTPU, a native PyTorch back end f... by sroussey on Tuesday October 20, 2026 @02:51PM (#50170918)
That explains the 5x speed increases huggingface mentioned for Mac using metal.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by mathisfun123 on Tuesday October 20, 2026 @04:03PM (#50171881)
Link?
Re:Google open-sources TorchTPU, a native PyTorch back end f... by nico on Tuesday October 20, 2026 @03:33PM (#50171527)
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?
Re:Google open-sources TorchTPU, a native PyTorch back end f... by cavisne on Tuesday October 20, 2026 @03:44PM (#50171644)
You can rent them on cloud + they sell directly to Anthropic. Its pretty well known that Anthropic is a heavy user of TPU's (along with Trainium & GPU's)
Re:Google open-sources TorchTPU, a native PyTorch back end f... by teaearlgraycold on Tuesday October 20, 2026 @03:43PM (#50171630)
They sell IoT-grade ones
Re:Google open-sources TorchTPU, a native PyTorch back end f... by snake_doc on Tuesday October 20, 2026 @09:20PM (#50174910)
Re:Google open-sources TorchTPU, a native PyTorch back end f... by atty on Tuesday October 20, 2026 @11:57PM (#50175919)
Google is selling TPUs to enterprise clients for on prem deployments. We’ve evaluated them but decided to stick with Nvidia, at least for one more generation. We’re a relatively large user of GCP.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by fancyfredbot on Tuesday October 20, 2026 @02:22PM (#50170486)
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.

Re:Google open-sources TorchTPU, a native PyTorch back end f... by ismailmaj on Tuesday October 20, 2026 @03:19PM (#50171330)
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?

Re:Google open-sources TorchTPU, a native PyTorch back end f... by Lucasoato on Tuesday October 20, 2026 @12:48PM (#50169173)
Are there TPUs outside Google Cloud?
Re:Google open-sources TorchTPU, a native PyTorch back end f... by boulos on Tuesday October 20, 2026 @01:08PM (#50169474)
Yes. Google is famously selling TPUs on prem to big customers (e.g., Anthropic) and Crux AI is now a venture to do just that.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by sschueller on Tuesday October 20, 2026 @12:55PM (#50169268)
Google coral USB and m.2 devices are TPUs.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by Archit3ch on Tuesday October 20, 2026 @02:27PM (#50170575)
"Have you seen any software run on Coral TPUs?"

"No, but I have seen the house that the project built, and it is terrific." ~ some Google director

Re:Google open-sources TorchTPU, a native PyTorch back end f... by charleslmunger on Tuesday October 20, 2026 @04:39PM (#50172363)
https://frigate.video

Is a pretty successful open source NVR that used corals when it launched (and many people still use them with it)

Re:Google open-sources TorchTPU, a native PyTorch back end f... by qlte on Wednesday October 21, 2026 @03:02AM (#50177051)
Yeah I have one for Frigate on a Raspberry Pi. Has worked well but definitely showing its age performance wise. I got a cheap mini PC (right before prices like quadrupled) which has about the same performance on an iGPU). I wouldn't be surprised if Frigate was the funnel for like 90% of Coral purchases or more.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by boulos on Tuesday October 20, 2026 @01:07PM (#50169459)
Only in marketing name. They're not related otherwise.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by goda90 on Tuesday October 20, 2026 @01:55PM (#50170123)
Maybe just marketing, but they call the NPUs in their Pixel phones "TPUs"
Re:Google open-sources TorchTPU, a native PyTorch back end f... by fulafel on Wednesday October 21, 2026 @01:37AM (#50176476)
TPUs are really matrix processors (they process 2d tensors right?), it's technically fair to call most matrix processors TPUs.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by westurner on Tuesday October 20, 2026 @12:55PM (#50169273)
TPU / neural compute accelerator / Math coprocessor;

"OpenTPU: Open-Source Reimplementation of Google Tensor Processing Unit (TPU)" https://news.ycombinator.com/item?id=44111452

From https://news.ycombinator.com/item?id=45848676 :

> How does Cerebras WSE-3 with 44GB of 'L2' on-chip SRAM compare to Google's TPUs, Tesla's TPUs, NorthPole, Groq LPU, Tenstorrent's, and AMD's NPU designs?

There are TPUs made out of silicon carbide with standard fab.

There is in-flight tensor processing with photons, but that's not quite a TPU.

Re:Google open-sources TorchTPU, a native PyTorch back end f... by hackernudes on Tuesday October 20, 2026 @01:59PM (#50170168)
I think the commenter was asking if Google sells TPU chip hardware to use outside of Google's own infrastructure and cloud.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by 23agt on Tuesday October 20, 2026 @01:27PM (#50169752)
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.

Re:Google open-sources TorchTPU, a native PyTorch back end f... by airza on Tuesday October 20, 2026 @01:33PM (#50169820)
it doesn't seem like there's an abundance of shovels coming anytime soon
Re:Google open-sources TorchTPU, a native PyTorch back end f... by juleiie on Tuesday October 20, 2026 @04:30PM (#50172225)
Shovels and gold aren’t the problem, humans are the problem

either AI achieves hyper-efficiency and triggers systematic economic demand destruction due to unemployment, or it underperforms and sparks a massive tech-sector asset bubble burst

If AI is as good as valuations - it crashes. If AI isn’t as good, it crashes. all eggs in one basket economy.

Every future leads to a massive financial shitstorm

AI is priced in as a technology to end all markets. The last bull market you will ever see.

It’s a miracle markets didn’t notice this “damned if you succeed damned if you don’t” type of inevitable checkmate

I have been building a bunker

Re:Google open-sources TorchTPU, a native PyTorch back end f... by bsder on Tuesday October 20, 2026 @08:17PM (#50174480)
> It’s a miracle markets didn’t notice this “damned if you succeed damned if you don’t” type of inevitable checkmate

They learned from 2008 that Too Big Too Fail(tm) means the government bails you out.

This was the problem with not OBLITERATING the banks in 2008. Had the US government taken over the banks and wiped the investors, there would be more pushback right now.

Re:Google open-sources TorchTPU, a native PyTorch back end f... by omgwtfbyobbq on Tuesday October 20, 2026 @05:32PM (#50172986)
It's possible. If goog's AI overview is accurate, it seems somewhat similar to speculative euphoria seen in the past (eg companies involved in electricity prior to the great depression). It's larger (currently 35% for AI versus 25% for electric/related cos), I'm guessing because of more credit. At the same time, the housing component of the market prior to the GFC was about 23%, and it wasn't as economically damaging as the GD, so hopefully the AI bubble unwinding won't be as bad.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by esseph on Tuesday October 20, 2026 @05:50PM (#50173186)
> so hopefully the AI bubble unwinding won't be as bad

It's currently larger than the 1880s railroad boom.

"Overbuilding and speculative debt ultimately culminated in the Panic of 1893, plunging the nation into a severe economic depression."

Re:Google open-sources TorchTPU, a native PyTorch back end f... by omgwtfbyobbq on Tuesday October 20, 2026 @06:56PM (#50173835)
Not according to Goog's. The railroad bubble of the 1880s was supposedly 60+% of the market. It may not be accurate though.

"During the 1880s, private railroad enterprises represented over 60% of the total capitalization of the entire U.S. stock market, peaking at 63% in 1881 and 1885."

Re:Google open-sources TorchTPU, a native PyTorch back end f... by esseph on Wednesday October 21, 2026 @11:55AM (#50183101)
AI spending is significant more in inflation adjusted dollars. (Roughly 550 Bn in modern dollars). We're projecting trillions in AI spend.

But as a % of total GDP, you are correct.

Re:Google open-sources TorchTPU, a native PyTorch back end f... by omgwtfbyobbq on Wednesday October 21, 2026 @05:26PM (#50187885)
I asked Goog's for enterprise value of the sector as a percentage of total market in all 4 situations, so it's not exact because enterprise value isn't available back to the 1800s, but it should be ballpark assuming the overall picture is accurate.

In terms of infra spending as a percent of GDP, 1880s railroads were at 5-6% at peak while AI is estimated to reach 3+% in a year or two.

Re:Google open-sources TorchTPU, a native PyTorch back end f... by tjwebbnorfolk on Tuesday October 20, 2026 @06:52PM (#50173788)
Everyone says there's a massive compute shortage. But where is it showing up? For whom is there a shortage?

Every time I ask Claude to do something, it does it just fine. I never have to wait in any sort of queue that would be common in other kinds of shortages.

Re:Google open-sources TorchTPU, a native PyTorch back end f... by eitally on Tuesday October 20, 2026 @07:28PM (#50174111)
It's showing up for enterprises (and ISVs) who need capacity from cloud vendors and can't get it. Sometimes there are some GPU shapes available, but not always what you either need or want to pay for (e.g. a TPU vs a V100 vs a B300), and also not always in spot instances (frequently not).

The hyperscalers are all so oversubscribed that they require commits from enterprises so if you only have a transient need you might be out of luck [from your vendor of choice]. Additionally, not all GPU service providers are equal. Semi Analysis has some great articles explaining why.

Re:Google open-sources TorchTPU, a native PyTorch back end f... by octoberfranklin on Tuesday October 20, 2026 @09:34PM (#50174995)
But where is it showing up?

Pricing -- exactly where shortages show up in a free market economy.

Re:Google open-sources TorchTPU, a native PyTorch back end f... by tjwebbnorfolk on Wednesday October 21, 2026 @11:23AM (#50182576)
Right, but pricing is for the marginal unit of hardware.

what I'm wondering: is there a shortage of compute in the current installed base? Why isn't the installed base of compute increasing drastically in price, and/or forcing me to wait in a queue to use it?

Re:Google open-sources TorchTPU, a native PyTorch back end f... by octoberfranklin on Wednesday October 21, 2026 @07:55PM (#50189258)
Why isn't the installed base of compute increasing drastically in price

Have you looked at RAM prices lately?

Re:Google open-sources TorchTPU, a native PyTorch back end f... by tjwebbnorfolk on Sunday October 25, 2026 @12:26PM (#50233286)
That's not the installed base. If you're buying RAM, that RAM isn't installed yet.

Installed base is existing computers that already have RAM. Why aren't VM rental prices going way up?

Re:Google open-sources TorchTPU, a native PyTorch back end f... by tjwebbnorfolk on Sunday October 25, 2026 @12:28PM (#50233302)
Right, but also shouldn't it be showing up in rationing as well? i.e. why am I not in a 4-hour-long queue waiting for Claude to respond?
Re:Google open-sources TorchTPU, a native PyTorch back end f... by octoberfranklin on Monday October 26, 2026 @11:07AM (#50245056)
facepalm

No, rationing is what you get in command/centralized economies like the USSR. In free market economies you get price explosion, like we're seeing right now with DRAM wafers.

Re:Google open-sources TorchTPU, a native PyTorch back end f... by airza on Wednesday October 21, 2026 @10:02AM (#50181154)
have you tried to purchase a device with RAM or a GPU in the last two years?
Re:Google open-sources TorchTPU, a native PyTorch back end f... by tjwebbnorfolk on Wednesday October 21, 2026 @11:22AM (#50182553)
Yes it's all very expensive. But is there not enough compute in general? Or is there not enough extra compute for me personally where I want it ?

like, I can rent a VM from AWS for the same price as 5 years ago...

Re:Google open-sources TorchTPU, a native PyTorch back end f... by bornfreddy on Wednesday October 21, 2026 @11:09AM (#50182335)
I mean there is an abundance of shovels - but everyone wants to dig with excavators, and there's a shortage of those.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by tuna74 on Tuesday October 20, 2026 @03:28PM (#50171456)
Is the Linux driver available?
Re:Google open-sources TorchTPU, a native PyTorch back end f... by yipinwong on Tuesday October 20, 2026 @01:07PM (#50169460)
Dumb question from non-AI engineer.

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

Re:Google open-sources TorchTPU, a native PyTorch back end f... by bitpush on Tuesday October 20, 2026 @01:15PM (#50169570)
I dont think big companies like Google react to news like that. Very likely this was meant to go out on Tuesday for weeks now. And it just happened to have coincided with another news.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by JacobAsmuth on Tuesday October 20, 2026 @01:58PM (#50170156)
Yeah they read the news this morning and whipped up a quick Torch->XLA translation layer and coordinated with four different inference engines (who also all wrote their own blog posts) and shoved this out the door quick to stay on top of the news cycle
Re:Google open-sources TorchTPU, a native PyTorch back end f... by parthdesai on Tuesday October 20, 2026 @02:11PM (#50170339)
Why be rude on purpose? OP clearly prefaced by saying it was a dumb question
Re:Google open-sources TorchTPU, a native PyTorch back end f... by caaqil on Tuesday October 20, 2026 @02:41PM (#50170768)
Welcome to the internet, great to have you here.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by luckydata on Tuesday October 20, 2026 @05:59PM (#50173270)
because the question was incredibly stupid and deserved some sarcasm
Re:Google open-sources TorchTPU, a native PyTorch back end f... by mathisfun123 on Tuesday October 20, 2026 @02:01PM (#50170195)
they're releasing this today because the pytorch conf is this week...
Re:Google open-sources TorchTPU, a native PyTorch back end f... by geremiiah on Tuesday October 20, 2026 @02:24PM (#50170515)
No, these are two very different objectives.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by Tanjreeve on Tuesday October 20, 2026 @01:20PM (#50169637)
They do different things and sit in different parts of the stack (training vs inference). TorchTPU seems more like a play to get more people familiar with Googles TPU chips as they want to sell them to people outside of google.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by beardedetim on Tuesday October 20, 2026 @02:22PM (#50170478)
Yeah I'd assume this is to allow us to train on their TPUs with torch instead of needing to rent a GPU
Re:Google open-sources TorchTPU, a native PyTorch back end f... by yipinwong on Tuesday October 20, 2026 @02:57PM (#50170998)
ty. that helps a lot and know where to research next.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by mathisfun123 on Tuesday October 20, 2026 @01:47PM (#50170003)
lol no this is a replacement for https://github.com/pytorch/xla
Re:Google open-sources TorchTPU, a native PyTorch back end f... by 0x00cl on Tuesday October 20, 2026 @08:49PM (#50174713)
We might never know, but they probably didn't read the news and started working to get this out, they were probably working on it for a while now.

Personally the question is if they are "saving" other projects or news so they can be released in case a competitor makes a move? Who knows if this was ready days or weeks ago and were waiting on someone to drop some news, so they can shadow those news.

Re:Google open-sources TorchTPU, a native PyTorch back end f... by nelaggy on Tuesday October 20, 2026 @01:50PM (#50170052)
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?
Re:Google open-sources TorchTPU, a native PyTorch back end f... by badrequest on Tuesday October 20, 2026 @02:10PM (#50170318)
hardware has always been the moat in the AI space
Re:Google open-sources TorchTPU, a native PyTorch back end f... by AnotherGoodName on Tuesday October 20, 2026 @03:15PM (#50171275)
Yeah a lot of people bought into the idea of CUDA being a moat but we’re well past the point where it’s pretty trivial to jump to another equivalent stack on the software side.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by jacquesm on Tuesday October 20, 2026 @03:29PM (#50171463)
Those are the two main interfaces: the inference hardware and the endpoints. Endpoints are pretty much standardized to the point that you can switch them out in seconds. CUDA still has an edge but every release like this chips away at that. Even so, TPUs are unobtanium so while I appreciate Google's effort here I also recognize that it is not something that will be useful to me.

In that sense AMD is still better positioned to give NVidia a run for their money but given the family connection and AMD's underdog status I am not very hopeful that we'll see meaningful competition in inference hardware any time soon.

NVidia is a much harder nut to crack than OpenAI or Anthropic. They are easily a decade ahead of viable competition. Check out the GB300 if you don't believe that.

Re:Google open-sources TorchTPU, a native PyTorch back end f... by nixon_why69 on Wednesday October 21, 2026 @12:39AM (#50176144)
A decade is a long time and the highest end isn't the only market segment.

High bandwidth flash will be here soon, I bet someone makes an economical competitor thats not quite as fast but much cheaper.

Re:Google open-sources TorchTPU, a native PyTorch back end f... by kavalg on Wednesday October 21, 2026 @02:46AM (#50176935)
While I have the same sentiment, I am also curious why a competitor can't accelerate their developments with AI. In the end, hardware development is to a large extent also software development. Is it the patents that NVIDIA has that are preventing this? Or maybe some trade secrets? AFAIK the NVIDIA hardware is manufactured entirely by TSMC.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by lp92 on Wednesday October 21, 2026 @12:39PM (#50183809)
Google does provide TPUs to big players like Anthropic and other large enterprise customers for on prem deploymens. That's their target just like those the GB300 is aimed at.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by coredog64 on Tuesday October 20, 2026 @03:58PM (#50171812)
Even if it's allegedly possible, the devil has always been in the details. It always sucks to find out that there's some quirk of your model architecture that breaks an implicit assumption that underpins the export process.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by AndrewKemendo on Tuesday October 20, 2026 @03:02PM (#50171071)
Will always be true
Re:Google open-sources TorchTPU, a native PyTorch back end f... by JacobAsmuth on Tuesday October 20, 2026 @01:57PM (#50170138)
Products can also be a moat.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by amelius on Tuesday October 20, 2026 @06:41PM (#50173682)
Soon to be written by AI anyway, so they might as well release it now.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by saucesaft on Tuesday October 20, 2026 @07:21PM (#50174047)
bad news for jax
Re:Google open-sources TorchTPU, a native PyTorch back end f... by jax_bro on Tuesday October 20, 2026 @07:54PM (#50174315)
not really!
Re:Google open-sources TorchTPU, a native PyTorch back end f... by jart on Wednesday October 21, 2026 @01:40AM (#50176499)
Let me guess, they gave you something better and it's eating you up inside? That's the way your hard-core Commie works.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by saagarjha on Wednesday October 21, 2026 @10:05AM (#50181205)
wat
Re:Google open-sources TorchTPU, a native PyTorch back end f... by jart on Wednesday October 21, 2026 @11:53AM (#50183061)
Re:Google open-sources TorchTPU, a native PyTorch back end f... by jax_bro on Wednesday October 21, 2026 @04:08PM (#50186902)
what does this mean!
Re:Google open-sources TorchTPU, a native PyTorch back end f... by frobisher on Wednesday October 21, 2026 @02:31AM (#50176840)
Question: does this make JAX obsolete? I was having fun learning it!
Re:Google open-sources TorchTPU, a native PyTorch back end f... by csvance on Wednesday October 21, 2026 @08:08AM (#50179534)
It makes PyTorch work nicely on TPU, which is impressive mostly because of how unstructured PyTorch is with its conventions. From a compiler standpoint JAX is much easier to work with since its all pure functions. This means that JAX also has access to fairly cutting edge compiler tech like Enzyme which is quite relevant for scientific machine learning work.
Re:Google open-sources TorchTPU, a native PyTorch back end f... by g58892881 on Wednesday October 21, 2026 @02:37AM (#50176870)
great! where can I buy a TPU?
Re:Google open-sources TorchTPU, a native PyTorch back end f... by ranger_danger on Wednesday October 21, 2026 @08:53AM (#50180082)
How about open-sourcing the proprietary blackbox Titan security chip, that you pinky-promised to open but never did?
Re:Google open-sources TorchTPU, a native PyTorch back end f... by tingkuanpei on Wednesday October 21, 2026 @08:58AM (#50180165)
Very interesting work. Google should have done this a long time ago. PyTorch's ecosystem is just so rich.