2 ms·
Sounds good, but my main question is: is this a fork, or a new backend they're building in (like MPS)?
by Reubend 6mo ago
Sounds good, but my main question is: is this a fork, or a new backend they're building in (like MPS)?
- musebox35 6mo agoI 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.
- yu3zhou4 6mo agoThey write that they use PrivateUse1, so it’s a custom out-of-tree backend