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.
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.