jax-js: JAX in JavaScript – ML library for the web, running on WebGPU & Wasm

https://myslop.shop/a/jax-js