Sign in
← Back to search
github Actively maintained

jax-ml/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

3 awesome lists

Quick read

Stars
36,252
Forks
3,752
Open issues
2,515
Commits
38,116

Activity and growth

Latest capture 2026-09-04 04:17

Stars · last 7 days
+30 +0.1%
Commits · last 7 days
+108 +0.3%
Stars since tracking
+580
Stored snapshots
101

Classification

Topics

#jax

Technology stack

pytest

Metadata

Language
Python
License
Apache-2.0
Default branch
main
Created
2018-10-25
First commit
2018-11-18
Last pushed
2026-09-04
GitHub updated
2026-09-04
Last synced
2026-09-04 04:17
Stack scanned
2026-09-04 04:17
Archived
No
Sign in to like jax-ml/jax
README Open website View on GitHub

AI development signals

0 paths

Agent instructions and tool configuration found in this repository.

No config files detected.

Growth history

Tracked growth

101 observed captures since 2026-05-22. Observed captures are shown by default.

Stars from first capture +580

Chart data

Observed captures only

Time horizon

All tracked data

Custom date range

Stars history

Observed snapshots

Commits history

Observed snapshots

Similar repositories

Nearest indexed repositories by embedding similarity.

patrick-kidger/equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

2,936 stars
Python 1 awesome list

patrick-kidger/diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

2,078 stars
Python 1 awesome list

google-deepmind/optax

Optax is a gradient processing and optimization library for JAX.

2,324 stars
Python 1 awesome list

google/flax

Flax is a neural network library for JAX that is designed for flexibility.

7,285 stars
Jupyter Notebook 2 awesome lists

facebook/Ax

Adaptive Experimentation Platform

2,790 stars
Python 1 awesome list