ekzhang/jax-js
JAX in JavaScript – ML library for the web, running on WebGPU & Wasm
733 stars. Actively maintained with 20 commits in the last 30 days.
Stars
733
Forks
38
Language
TypeScript
License
MIT
Category
Last pushed
Mar 13, 2026
Commits (30d)
20
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
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
google/grain
Library for reading and processing ML training data.
extropic-ai/thrml
Thermodynamic Hypergraphical Model Library in JAX