jax-js Brings JAX-Style ML Kernels To The Browser

Published by

on •1 min read

jax-js Brings JAX-Style ML Kernels To The Browser

A pure JavaScript ML framework that runs on WebAssembly and WebGPU. jax-js is Eric Zhang’s attempt to make the browser feel a bit more like a tiny JAX lab.

Showcase WebGPU WASM AI/ML Compute Shaders Math Interactive Learning

Learn more