[fisf]

That's fine and dandy, until you realize that Jax only has a limited amount of backends. E.g. rocm support is still experimental.

Somebody has to build those optimized backends -- it's not just a matter of people picking the wrong stack.

[nerpderp82]

I just looked at Jax and XLA, it is odd to me that they aren't targeting SPIR-V directly.