jax-js REPL

Try out jax-js. 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 @jax-js/jax

Examples