top | item 46314231

Show HN: jax-js, an ML library and compiler for the web

4 points| ekzhang | 2 months ago |ekzhang.substack.com

1 comment

order

ekzhang|2 months ago

Wanted to share this library that I'm open-sourcing now after working on it for the past year.

You write code like in JAX/NumPy, but it’s fully interactive on the frontend and compiles down to shaders on the user’s GPU (with WebGPU). So far I’ve used it for purely frontend-only ML demos. https://jax-js.com/mobileclip