jax-js REPL

Try out jax-js. Numerical and GPU computing for the web!

The library has NumPy and JAX-like APIs in the browser, on Wasm and WebGPU, with JIT compilation.

npm i @jax-js/jax

Examples