3 ms·
But then they would have to compile the jit-optimized XLA to C. Do you know if that’s straightforwardly doable?
by 6gvONxR4sf7o 3y ago
But then they would have to compile the jit-optimized XLA to C. Do you know if that’s straightforwardly doable?
- fho 3y agoHmm ... Good question. I would assume that JAX and XLA is based on LLVM (like everything else these days) and that you could probably grab the IR at some point to compile it to your target.