alt Hacker NewsSounds good, but my main question is: is this a fork, or a new backend they're building in (like MPS)?
Replies
musebox35 • today at 4:28 AM
I attended the related session at Next’26 yesterday. From my understanding it is a new backend and they will release the torch tpu source on github in one or two months. It will not support all ops initially but they are moving fast. Still for a while torchax is mature enough to run torch models on tpus by translating to jax.
They write that they use PrivateUse1, so it’s a custom out-of-tree backend