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

Show HN: jax-js, an ML library and compiler for the web(ekzhang.substack.com)

4 points by ekzhang 151 days ago | 1 comment

ekzhang 151 days 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

ekzhang 151 days 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