4 ms·

jax in the browser on WebGPU next please?

• scroogey 2mo ago
  https://jax-js.com/ https://jax-js.com/
  • hessammehr 2mo ago
    I’m aware of Jax-js. It’s a different beast: an [incomplete last I checked] port of jax for JS, not something you can run your, e.g. jax-dependent marimo notebook or pyodide against
• momojo 2mo ago
  Why not use Numba? In my mental model they achieve the same win, generating GIL-free, bytecode-level perf. I'm not super familiar with JAX though.