@jax-js/jax 0.1.10 → 0.1.12
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +7 -2
- package/dist/{backend-Ctqs8la1.js → backend-DI-V78Rk.js} +732 -21
- package/dist/{backend-DMauYnfl.cjs → backend-x-6vqzIM.cjs} +737 -20
- package/dist/index.cjs +372 -20
- package/dist/index.d.cts +172 -4
- package/dist/index.d.ts +172 -4
- package/dist/index.js +372 -21
- package/dist/{webgl-CvQ1QBX1.js → webgl-BhsnpeB0.js} +7 -1
- package/dist/{webgl-kvVt7-T7.cjs → webgl-CD3WK_Me.cjs} +7 -1
- package/dist/{webgpu-v_W_-oKw.js → webgpu-C2kLdkUh.js} +299 -149
- package/dist/{webgpu-DMSx7a6M.cjs → webgpu-C4S8Uq9e.cjs} +299 -149
- package/package.json +1 -1
package/README.md
CHANGED
|
@@ -63,6 +63,7 @@ of late 2025.
|
|
|
63
63
|
|
|
64
64
|
Community usage:
|
|
65
65
|
|
|
66
|
+
- [**g9-jaxjs**: Automatically interactive graphics with forward-mode AD](https://srush.github.io/g9jax/)
|
|
66
67
|
- [**autoresearch-webgpu**: autoresesarch, in the browser](https://autoresearch.lucasgelfond.online/)
|
|
67
68
|
- [**tanh.xyz**: Interactive ML visualizations](https://tanh.xyz/)
|
|
68
69
|
- [**jax-js-bayes**: Declarative Bayesian modeling library](https://github.com/StefanSko/jax-js-bayes)
|
|
@@ -72,9 +73,12 @@ Demos on the jax-js website:
|
|
|
72
73
|
- [Training neural networks on MNIST](https://jax-js.com/mnist)
|
|
73
74
|
- [Voice cloning: Kyutai Pocket TTS](https://jax-js.com/tts)
|
|
74
75
|
- [CLIP embeddings for books in-browser](https://jax-js.com/mobileclip)
|
|
76
|
+
- [Object detection: D-FINE (ONNX)](https://jax-js.com/d-fine)
|
|
75
77
|
- [Object detection: DETR ResNet-50 (ONNX)](https://jax-js.com/detr-resnet-50)
|
|
78
|
+
- [Fluid simulation (Navier-Stokes)](https://jax-js.com/fluid-sim)
|
|
76
79
|
- [In-browser REPL](https://jax-js.com/repl)
|
|
77
80
|
- [Matmul benchmark](https://jax-js.com/bench/matmul)
|
|
81
|
+
- [Matvec benchmark](https://jax-js.com/bench/matvec)
|
|
78
82
|
- [Conv2d benchmark](https://jax-js.com/bench/conv2d)
|
|
79
83
|
- [Mandelbrot set](https://jax-js.com/mandelbrot)
|
|
80
84
|
|
|
@@ -310,7 +314,9 @@ and determines how to execute compiled operations on them.
|
|
|
310
314
|
There are currently 4 devices in jax-js:
|
|
311
315
|
|
|
312
316
|
- `cpu`: Slow, interpreted JS, only meant for debugging.
|
|
313
|
-
- `wasm`: [WebAssembly](https://webassembly.org/),
|
|
317
|
+
- `wasm`: [WebAssembly](https://webassembly.org/), multi-threaded when
|
|
318
|
+
[`SharedArrayBuffer`](https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/SharedArrayBuffer)
|
|
319
|
+
is available.
|
|
314
320
|
- `webgpu`: [WebGPU](https://developer.mozilla.org/en-US/docs/Web/API/WebGPU_API), available on
|
|
315
321
|
[supported browsers](https://caniuse.com/webgpu) (Chrome, Firefox, Safari, iOS).
|
|
316
322
|
- `webgl`: [WebGL2](https://developer.mozilla.org/en-US/docs/Web/API/WebGL2RenderingContext), via
|
|
@@ -419,7 +425,6 @@ Contributions are welcomed! Some fruitful areas to look into:
|
|
|
419
425
|
- Adding support for more JAX functions and operations, see [compatibility table](./FEATURES.md).
|
|
420
426
|
- Improving performance of the WebGPU and Wasm runtimes, generating better kernels, and using SIMD
|
|
421
427
|
and multithreading. (Even single-threaded Wasm could be ~20x faster.)
|
|
422
|
-
- Helping the JIT compiler to fuse operations in more cases, like `tanh` branches.
|
|
423
428
|
- Making a fast transformer inference engine, comparing against onnxruntime-web.
|
|
424
429
|
|
|
425
430
|
You may join our [Discord server](https://discord.gg/BW6YsCd4Tf) and chat with the community.
|