4 ms·
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 o
by fisf 3y ago
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 3y agoI just looked at Jax and XLA, it is odd to me that they aren't targeting SPIR-V directly.