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.

Open REPL

Run code to see output here.

Learn more

GitHub Repository

Get started with jax-js and check out the tutorial.

MNIST Training

Demo of training a neural network on MNIST.

MobileCLIP2 Inference

Generate embeddings for books and search them in real time.