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.
Live examples
Interactive demos for local AI models, simulations, geometry processing, and numerical methods.
Gemma 3 Chat
Run a small language model locally in your browser, with GPU and CPU-based inference.
Neural Cellular Automata
Grow image patterns with a tiny differentiable automaton.
%20scale(1.2)%20translate(-185%20-91)'%3e%3cpath%20d='M96%20108%20L134%2057%20L190%2042%20L252%2063%20L278%20112%20L236%20140%20L166%20132%20Z'%20fill='url(%23heat-abstract-fill)'%20opacity='0.95'%20/%3e%3cg%20fill='none'%20stroke='%23f7f2df'%20stroke-linejoin='round'%20stroke-width='1.4'%20opacity='0.58'%3e%3cpath%20d='M96%20108%20L134%2057%20L190%2042%20L252%2063%20L278%20112%20L236%20140%20L166%20132%20Z'%20/%3e%3cpath%20d='M134%2057%20L166%20132%20M190%2042%20L166%20132%20M252%2063%20L166%20132'%20/%3e%3cpath%20d='M96%20108%20L190%2042%20M134%2057%20L252%2063%20M252%2063%20L236%20140%20M166%20132%20L278%20112'%20/%3e%3c/g%3e%3cg%20fill='none'%20stroke='%23fff7c2'%20stroke-linecap='round'%20stroke-width='2.4'%3e%3cellipse%20cx='157'%20cy='96'%20rx='18'%20ry='13'%20transform='rotate(-18%20157%2096)'%20/%3e%3cellipse%20cx='157'%20cy='96'%20rx='39'%20ry='27'%20transform='rotate(-18%20157%2096)'%20opacity='0.92'%20/%3e%3cellipse%20cx='157'%20cy='96'%20rx='63'%20ry='41'%20transform='rotate(-18%20157%2096)'%20opacity='0.76'%20/%3e%3cellipse%20cx='157'%20cy='96'%20rx='87'%20ry='55'%20transform='rotate(-18%20157%2096)'%20opacity='0.52'%20/%3e%3c/g%3e%3ccircle%20cx='157'%20cy='96'%20r='6'%20fill='%23111827'%20/%3e%3ccircle%20cx='157'%20cy='96'%20r='2.2'%20fill='%23fff7c2'%20/%3e%3c/g%3e%3c/svg%3e)
Heat Method Geodesics
Compute surface distances on triangle meshes with dense solves.
'%3e%3crect%20x='0'%20y='58'%20width='14'%20height='18'%20rx='2'%20fill='%23244f86'%20opacity='0.88'%20/%3e%3crect%20x='22'%20y='38'%20width='14'%20height='38'%20rx='2'%20fill='%231fb6a6'%20opacity='0.88'%20/%3e%3crect%20x='44'%20y='20'%20width='14'%20height='56'%20rx='2'%20fill='%23f8c44f'%20opacity='0.88'%20/%3e%3crect%20x='66'%20y='48'%20width='14'%20height='28'%20rx='2'%20fill='%23244f86'%20opacity='0.88'%20/%3e%3crect%20x='88'%20y='4'%20width='14'%20height='72'%20rx='2'%20fill='%23f8c44f'%20opacity='0.88'%20/%3e%3crect%20x='110'%20y='30'%20width='14'%20height='46'%20rx='2'%20fill='%231fb6a6'%20opacity='0.88'%20/%3e%3crect%20x='132'%20y='46'%20width='14'%20height='30'%20rx='2'%20fill='%23244f86'%20opacity='0.88'%20/%3e%3crect%20x='154'%20y='10'%20width='14'%20height='66'%20rx='2'%20fill='%23f8c44f'%20opacity='0.88'%20/%3e%3crect%20x='176'%20y='36'%20width='14'%20height='40'%20rx='2'%20fill='%231fb6a6'%20opacity='0.88'%20/%3e%3crect%20x='198'%20y='54'%20width='14'%20height='22'%20rx='2'%20fill='%23244f86'%20opacity='0.88'%20/%3e%3crect%20x='220'%20y='24'%20width='14'%20height='52'%20rx='2'%20fill='%23f8c44f'%20opacity='0.88'%20/%3e%3crect%20x='242'%20y='42'%20width='14'%20height='34'%20rx='2'%20fill='%23244f86'%20opacity='0.88'%20/%3e%3c/g%3e%3cpath%20d='M40%20105%20C58%2083%2073%20126%2091%20104%20C109%2082%20122%20126%20141%20104%20C160%2082%20176%20126%20195%20104%20C214%2084%20232%20124%20251%20105%20C272%2086%20295%20116%20320%2097'%20fill='none'%20stroke='%23d9f99d'%20stroke-linecap='round'%20stroke-width='3'%20/%3e%3cline%20x1='226'%20y1='32'%20x2='226'%20y2='140'%20stroke='%23f97316'%20stroke-width='2'%20opacity='0.95'%20/%3e%3c/svg%3e)
Whisper ASR
Transcribe audio locally with OpenAI Whisper.
Kyutai Pocket TTS
Voice cloning AI model that runs in your browser.
Fluid Simulation
Interactive Navier-Stokes flow with WebGPU kernels.
D-FINE Detection
Run an ONNX object detector locally with WebGPU.
'%3e%3crect%20x='2'%20y='2'%20width='118'%20height='118'%20rx='5'%20fill='%23172033'%20opacity='0.06'%20/%3e%3crect%20width='118'%20height='118'%20rx='5'%20fill='%23f7fafd'%20stroke='%23c8d5e2'%20stroke-width='1.5'%20/%3e%3cg%20stroke='%23c8d5e2'%20stroke-width='1.3'%20fill='none'%3e%3cpath%20d='M23.6%200%20V118%20M47.2%200%20V118%20M70.8%200%20V118%20M94.4%200%20V118'%20/%3e%3cpath%20d='M0%2023.6%20H118%20M0%2047.2%20H118%20M0%2070.8%20H118%20M0%2094.4%20H118'%20/%3e%3c/g%3e%3cg%20fill='%23d1d9e5'%3e%3crect%20x='26'%20y='4'%20width='15'%20height='15'%20rx='3'%20/%3e%3crect%20x='49'%20y='28'%20width='15'%20height='15'%20rx='3'%20/%3e%3crect%20x='75'%20y='52'%20width='15'%20height='15'%20rx='3'%20/%3e%3crect%20x='4'%20y='76'%20width='15'%20height='15'%20rx='3'%20/%3e%3crect%20x='100'%20y='6'%20width='13'%20height='13'%20rx='3'%20opacity='0.45'%20/%3e%3crect%20x='30'%20y='99'%20width='13'%20height='13'%20rx='3'%20opacity='0.38'%20/%3e%3c/g%3e%3cg%20fill='%23ff7115'%3e%3crect%20x='2'%20y='2'%20width='19.5'%20height='19.5'%20rx='3.2'%20/%3e%3crect%20x='25.6'%20y='25.6'%20width='19.5'%20height='19.5'%20rx='3.2'%20/%3e%3crect%20x='49.2'%20y='49.2'%20width='19.5'%20height='19.5'%20rx='3.2'%20/%3e%3crect%20x='72.8'%20y='72.8'%20width='19.5'%20height='19.5'%20rx='3.2'%20/%3e%3crect%20x='96.4'%20y='96.4'%20width='19.5'%20height='19.5'%20rx='3.2'%20/%3e%3c/g%3e%3c/g%3e%3c/svg%3e)
MobileCLIP2 Inference
Compute embeddings for book passages.
MNIST Training
Train a neural network on handwritten digits in-browser.
'%20fill='%2364748b'%20opacity='0.14'%20/%3e%3cg%20fill='none'%20stroke-linecap='round'%20stroke-width='3'%3e%3cpath%20d='M111%20121%20L282%2058'%20stroke='%23dc2626'%20/%3e%3cpath%20d='M111%20121%20L137%2039'%20stroke='%2316a34a'%20/%3e%3cpath%20d='M111%20121%20L75%2067'%20stroke='%232563eb'%20/%3e%3c/g%3e%3ccircle%20cx='111'%20cy='121'%20r='4'%20fill='%23111827'%20/%3e%3cg%20opacity='0.9'%3e%3ccircle%20cx='78'%20cy='124'%20r='3.2'%20fill='%239a8d22'%20/%3e%3ccircle%20cx='94'%20cy='109'%20r='3.2'%20fill='%235f9b42'%20/%3e%3ccircle%20cx='111'%20cy='116'%20r='3.2'%20fill='%23b37b22'%20/%3e%3ccircle%20cx='127'%20cy='96'%20r='3.2'%20fill='%235f9b42'%20/%3e%3ccircle%20cx='145'%20cy='102'%20r='3.2'%20fill='%239a8d22'%20/%3e%3ccircle%20cx='164'%20cy='88'%20r='3.2'%20fill='%235f9b42'%20/%3e%3ccircle%20cx='182'%20cy='91'%20r='3.2'%20fill='%23b37b22'%20/%3e%3ccircle%20cx='201'%20cy='79'%20r='3.2'%20fill='%235f9b42'%20/%3e%3ccircle%20cx='219'%20cy='70'%20r='3.2'%20fill='%239a8d22'%20/%3e%3ccircle%20cx='238'%20cy='74'%20r='3.2'%20fill='%235f9b42'%20/%3e%3ccircle%20cx='257'%20cy='61'%20r='3.2'%20fill='%23b37b22'%20/%3e%3ccircle%20cx='277'%20cy='67'%20r='3.2'%20fill='%235f9b42'%20/%3e%3ccircle%20cx='121'%20cy='130'%20r='3.2'%20fill='%235f9b42'%20/%3e%3ccircle%20cx='154'%20cy='121'%20r='3.2'%20fill='%23b37b22'%20/%3e%3ccircle%20cx='189'%20cy='113'%20r='3.2'%20fill='%235f9b42'%20/%3e%3ccircle%20cx='226'%20cy='99'%20r='3.2'%20fill='%239a8d22'%20/%3e%3c/g%3e%3c/svg%3e)
Principal Component Analysis
Explore components of a 3D point cloud.
Benchmarks
Compare jax-js kernels with other web ML libraries.