Open Source
Jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Our verdict
We have not reviewed Jax yet
No one at Enterprise DNA has checked this listing against its sources. The description comes from public listings gathered automatically. Any fact below shows where it came from and when it was last checked.
From the listing
About Jax
Jax is a Python library for composable transformations of numerical code built on NumPy. It enables automatic differentiation, vectorization, and JIT compilation to GPU and TPU hardware. Designed for high-performance scientific computing and machine learning workloads.
Best for
Researchers and engineers building custom numerical algorithms that need automatic differentiation and hardware acceleration.
Use cases
- Training neural networks with automatic differentiation on accelerators
- Vectorizing batch operations across arrays without explicit loops
- Compiling numerical algorithms to GPU/TPU for production inference
Notes
35,725 stars on GitHub. Last updated 2026-06-01. Licensed Apache-2.0.
Pros
- Functional programming model makes code composable and transformable
- Native GPU/TPU compilation with JIT removes manual optimization overhead
- NumPy-compatible API reduces learning curve for numerical Python developers
Cons
- Steeper learning curve than standard NumPy due to functional constraints and immutability
- Debugging transformed code is harder than debugging plain Python
- Ecosystem smaller than PyTorch or TensorFlow for pre-built models and utilities
Key facts
What we have checked
- Latest version
- jax-v0.12.0.1 (released 22 Oct 2026)
- GitHub stars
- 36,339
Each value shows when we last confirmed it and where it came from. Values we have not confirmed are left out, never guessed.
Stars and usage
How Jax is growing
35,725 to 36,339
Alternatives
Alternatives to Jax
| Record | Type | How we know | Checked |
|---|---|---|---|
| Apache MXNetAlternative to this | Open source | Named in the listing | Not dated |
| CandleAlternative to this | Open source | Named in the listing | Not dated |
| KerasAlternative to this | Open source | Named in the listing | Not dated |
| OneflowAlternative to this | Open source | Named in the listing | Not dated |
| PyTorchAlternative to this | Open source | Named in the listing | Not dated |
| TensorFlowAlternative to this | Open source | Named in the listing | Not dated |
| VectorFlowAlternative to this | Open source | Named in the listing | Not dated |
Compare
Compare Jax side by side
Sources
Where this comes from
- GitHubOfficial
- github api via nightly crawl, recovered from git history 20e874270887c1008f860ee0198e461a23c9e438Registry
A source shows what was published when we checked it. It does not certify security, performance or a working installation. Read our method.
Changelog
What changed
Released jax-v0.12.0.1
Released jax-v0.12.0
Released jax-v0.11.2
Released jax-v0.11.1
Released jax-v0.11.0
Released jax-v0.10.2
Released jax-v0.10.1
Released jax-v0.10.0
Released jax-v0.9.2
Released jax-v0.9.1
Open-source & AI alternatives
Swap-in tools that solve the same job. Weigh the trade-offs before you commit.
Apache MXNet
Community
Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more
Candle
Community
Minimalist ML framework for Rust
Keras
Community
Deep Learning for humans
Oneflow
Community
OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.
PyTorch
Community
Tensors and Dynamic neural networks in Python with strong GPU acceleration
TensorFlow
Community
An Open Source Machine Learning Framework for Everyone
VectorFlow
Community
A minimalist neural network library optimized for sparse data and single machine environments.
Pairs with
Other entries in the index that connect to this one. Click through to see the chain.
Awesome Tensor Compilers
Community
A list of awesome compiler projects and papers for tensor computation and deep learning.
Jupyter Notebooks
Community
Jupyter Interactive Notebook
NNI
Community
An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
tokenizers
Community
💥 Fast State-of-the-Art Tokenizers optimized for Research and Production
TVM
Community
Open Machine Learning Compiler Framework
Get the free Developer’s Field Guide
A 27-page field guide to the AI coding workflow with Claude. Claude Code, MCP servers, the prompt patterns that work, and what to delegate. Free.
Enter your work email. We send it straight over, plus a few short notes worth knowing. Unsubscribe any time.