Try out jax-js-nonconsuming. Numerical and GPU computing for the web!
The goal is having NumPy and JAX-like APIs in the browser, on Wasm or WebGPU — with JIT compilation.
npm i @hamk-uas/jax-js-nonconsuming