ekzhang/jax-js
JAX in JavaScript – ML library for the web, running on WebGPU & Wasm
This project helps developers build and run high-performance machine learning and numerical computing applications directly in web browsers. It takes mathematical operations and data (like numbers and arrays) in JavaScript, processes them efficiently using your device's CPU or GPU, and produces computed results or visual outputs for web-based tools. Web developers creating interactive data visualizations, AI-powered web apps, or scientific simulations are the primary users.
733 stars. Actively maintained with 19 commits in the last 30 days.
Use this if you are a web developer who needs to perform complex mathematical computations or run machine learning models directly within a web browser, leveraging the user's GPU for speed.
Not ideal if you are looking for a server-side machine learning solution or if your primary need is to simply load and run pre-trained ONNX models with minimal custom logic.
Stars
733
Forks
38
Language
TypeScript
License
MIT
Category
Last pushed
Mar 13, 2026
Commits (30d)
19
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/ekzhang/jax-js"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable....
google/grain
Library for reading and processing ML training data.
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/