jax-js is a machine learningan ML library and compiler for the web
High-performance WebGPU and WebAssembly kernels in JavaScript. Run neural networks, image algorithms, simulations, and numerical code, all JIT compiled in your browser.
Add jax-js to your project
Zero dependencies. All major browsers, with and in .
Matrix multiplication
Billions of floating-point operations (GFLOPs) per second
Running benchmark…
Try it out!
This is a live editor, the code is running in your browser.
Run code to see output here.
Learn more
GitHub Repository
Get started with jax-js and check out the tutorial.
REPL
Try out the library in this browser-based REPL.
API Reference
View the full API documentation.
Gemma 3 Chat
Run a small language model locally in your browser.
MobileCLIP2 Inference
Compute embeddings for book passages and search through them.
D-FINE Detection
Run an ONNX object detector locally with WebGPU.
Kyutai Pocket TTS
Voice cloning AI model that runs in your browser.
MNIST Training
Demo of training a neural network on MNIST.
Benchmarks
Micro-benchmarks of jax-js versus other libraries.