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
Why not use Numba? In my mental model they achieve the same win, generating GIL-free, bytecode-level perf.