web-ml

jax-js is an open-source ML library and compiler for the web, bringing JAX-like numerical computing to JavaScript with WebGPU and WebAssembly kernels, all running client-side.

Progress on jax-js includes new matmul benchmarks, faster real-time TTS demo, improved codegen, and an LLM demo running Gemma 3 270B. jax-js is an open-source web ML framework.