4 ms·
Funfact: you can probably JIT compile that using JAX for an easy performance gain.
by fho 3y ago
Funfact: you can probably JIT compile that using JAX for an easy performance gain.
- 6gvONxR4sf7o 3y agoBut 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.
- nephanth 3y agoNumba would probably do the job too