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 .
> npm install @jax-js/jax
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.
MobileCLIP2 Inference
Generate embeddings for books and search them in real time.
Kyutai Pocket TTS
Voice cloning AI model that runs in your browser.
MNIST Training
Demo of training a neural network on MNIST.