3 ms·
What was the estimate for how much time you guys took to translate the torch to Jax vs how much you spent on XLA?
by htrp 2y ago
What was the estimate for how much time you guys took to translate the torch to Jax vs how much you spent on XLA?
- felarof 2y agoIt took roughly 2-3 weeks to translate Torch to JAX, but I had past experience writing JAX from my time at Google. We spent nearly 4 weeks getting PyTorch XLA working on TPU. Hope that answers your question!