Data

Tools

Indexes

Applications

Experiments

Summary

An open API service for producing an overview of a list of open source projects.

https://github.com/jax-ml/jax

jax

Score: 38.26877383102379

Last synced: about 9 hours ago
JSON representation

Repository metadata:

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

Owner metadata:

Committers metadata

Last synced: about 24 hours ago

Total Commits: 26,023
Total Committers: 961
Avg Commits per committer: 27.079
Development Distribution Score (DDS): 0.882

Commits in past year: 6,788
Committers in past year: 299
Avg Commits per committer in past year: 22.702
Development Distribution Score (DDS) in past year: 0.874

Name Email Commits
Peter Hawkins p****s@g****m 3066
Jake VanderPlas j****p@g****m 2617
Matthew Johnson m****j@g****m 2312
Yash Katariya y****a@g****m 2076
jax authors g****n@g****m 1519
George Necula g****a@g****m 1201
Adam Paszke a****e@g****m 1051
Sergei Lebedev s****v@g****m 859
jax authors n****y@g****m 697
Roy Frostig f****g@g****m 627
Skye Wanderman-Milne s****m@g****m 496
Sharad Vikram s****v@g****m 434
Benjamin Chetioui b****i@g****m 378
Jake VanderPlas v****s@g****m 297
Dan Foreman-Mackey d****m@g****m 283
Allan Renucci a****i@g****m 280
Tomás Longeri t****i@g****m 223
Justin Fu j****u@g****m 223
Parker Schuh p****s@g****m 217
Daniel Suo d****o@g****m 186
Dimitar (Mitko) Asenov d****v@g****m 160
Stephan Hoyer s****r@g****m 145
Jevin Jiang j****g@g****m 141
Chris Jones c****j@g****m 140
Dougal d****n@g****m 132
Nitin Srinivasan s****n@g****m 125
rajasekharporeddy r****p@g****m 112
Lena Martens l****s@g****m 107
Qiao Zhang z****c@g****m 96
James Bradbury j****y@g****m 95
and 931 more...

Issue and Pull Request metadata

Last synced: about 14 hours ago

Total issues: 1,217
Total pull requests: 9,640
Average time to close issues: 8 months
Average time to close pull requests: 10 days
Total issue authors: 660
Total pull request authors: 261
Average comments per issue: 2.47
Average comments per pull request: 0.23
Merged pull request: 5,516
Bot issues: 62
Bot pull requests: 7,004

Past year issues: 373
Past year pull requests: 4,151
Past year average time to close issues: 8 days
Past year average time to close pull requests: 4 days
Past year issue authors: 228
Past year pull request authors: 135
Past year average comments per issue: 1.09
Past year average comments per pull request: 0.14
Past year merged pull request: 2,327
Past year bot issues: 20
Past year bot pull requests: 3,214

More stats: https://issues.ecosyste.ms/repositories/lookup?url=https://github.com/jax-ml/jax

Top Issue Authors

  • PhilipVinc (42)
  • copybara-service[bot] (42)
  • carlosgmartin (41)
  • ayaka14732 (24)
  • jakevdp (21)
  • github-actions[bot] (18)
  • patrick-kidger (17)
  • hawkinsp (14)
  • apiqwe (14)
  • sbodenstein (13)
  • gnecula (12)
  • vfdev-5 (12)
  • crusaderky (12)
  • mattjj (11)
  • NeilGirdhar (10)

Top Pull Request Authors

  • copybara-service[bot] (6,884)
  • jakevdp (546)
  • mattjj (214)
  • hawkinsp (188)
  • dfm (180)
  • gnecula (158)
  • dependabot[bot] (120)
  • vfdev-5 (67)
  • carlosgmartin (61)
  • rajasekharporeddy (50)
  • Ruturaj4 (47)
  • dougalm (40)
  • andportnoy (38)
  • MichaelHudgins (32)
  • Cjkkkk (30)

Top Issue Labels

  • bug (775)
  • enhancement (270)
  • documentation (27)
  • NVIDIA GPU (24)
  • pallas (24)
  • question (24)
  • P2 (eventual) (14)
  • P3 (no schedule) (13)
  • free threading (12)
  • pull ready (11)
  • build failed (10)
  • performance (9)
  • contributions welcome (9)
  • CI (8)
  • Apple GPU (Metal) plugin (6)
  • needs info (6)
  • XLA (6)
  • better_errors (5)
  • P1 (soon) (4)
  • cleanup (4)
  • CPU (4)
  • open (4)
  • good first issue (3)
  • python (2)
  • dependencies (2)
  • kokoro:force-run (2)
  • application (2)
  • type:performance (2)
  • AMD GPU (2)
  • benmarte (1)

Top Pull Request Labels

  • pull ready (1,848)
  • kokoro:force-run (635)
  • documentation (169)
  • dependencies (120)
  • python (82)
  • CI Optional GPU Presubmit (48)
  • github_actions (36)
  • CI Connection Halt - On Retry (20)
  • better_errors (14)
  • windows:force-run (8)
  • cla: yes (7)
  • free threading (4)
  • needs info (4)
  • skill issue (4)
  • enhancement (4)
  • bug (3)
  • CI (3)
  • AMD GPU (2)
  • CI Connection Halt - Always (2)
  • NVIDIA GPU (2)
  • pallas (2)
  • ruby (1)
  • build (1)
  • question (1)
  • CI Connection Halt - On Error (1)
  • Apple GPU (Metal) plugin (1)

Package metadata

nixpkgs-unstable: python313Packages.jax-cuda12-plugin

JAX Plugin for CUDA12

pypi.org: jax

Differentiate, compile, and transform Numpy code.

  • Homepage: https://github.com/jax-ml/jax
  • Documentation: https://jax.readthedocs.io/
  • Licenses: Apache-2.0
  • Latest release: 0.9.2 (published 20 days ago)
  • Last Synced: 2026-04-07T09:57:58.266Z (about 19 hours ago)
  • Versions: 190
  • Dependent Packages: 785
  • Dependent Repositories: 6,336
  • Downloads: 17,129,468 Last month
  • Docker Downloads: 1,084,879,634
  • Rankings:
    • Dependent packages count: 0.034%
    • Stargazers count: 0.079%
    • Dependent repos count: 0.12%
    • Downloads: 0.153%
    • Average: 0.184%
    • Forks count: 0.306%
    • Docker downloads count: 0.409%
  • Maintainers (9)
pypi.org: jaxlib

XLA library for JAX

  • Homepage: https://github.com/jax-ml/jax
  • Documentation: https://jaxlib.readthedocs.io/
  • Licenses: Apache-2.0
  • Latest release: 0.9.2 (published 20 days ago)
  • Last Synced: 2026-04-07T16:01:16.571Z (about 13 hours ago)
  • Versions: 81
  • Dependent Packages: 511
  • Dependent Repositories: 4,102
  • Downloads: 13,378,798 Last month
  • Docker Downloads: 32,935,086
  • Rankings:
    • Dependent packages count: 0.052%
    • Stargazers count: 0.079%
    • Dependent repos count: 0.155%
    • Average: 0.237%
    • Downloads: 0.242%
    • Forks count: 0.306%
    • Docker downloads count: 0.586%
  • Maintainers (9)
conda-forge.org: jax

  • Homepage: https://github.com/jax-ml/jax
  • Licenses: Apache-2.0
  • Latest release: 0.3.25 (published over 3 years ago)
  • Last Synced: 2026-03-25T18:08:14.239Z (13 days ago)
  • Versions: 90
  • Dependent Packages: 33
  • Dependent Repositories: 60
  • Downloads: 1,693,877 Total
  • Rankings:
    • Stargazers count: 1.517%
    • Dependent packages count: 2.071%
    • Forks count: 2.635%
    • Average: 2.698%
    • Dependent repos count: 4.571%
spack.io: py-jax

Differentiate, compile, and transform Numpy code. JAX is a Python library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and large-scale machine learning.

  • Homepage: https://github.com/jax-ml/jax
  • Licenses: []
  • Latest release: 0.9.2 (published 12 days ago)
  • Last Synced: 2026-04-05T19:02:15.689Z (2 days ago)
  • Versions: 55
  • Dependent Packages: 14
  • Dependent Repositories: 0
  • Rankings:
    • Dependent repos count: 0.0%
    • Stargazers count: 0.403%
    • Forks count: 1.848%
    • Average: 2.991%
    • Dependent packages count: 9.712%
  • Maintainers (1)
conda-forge.org: jaxlib

  • Homepage: http://github.com/jax-ml/jax
  • Licenses: Apache-2.0
  • Latest release: 0.3.25 (published over 3 years ago)
  • Last Synced: 2026-03-25T18:08:13.142Z (13 days ago)
  • Versions: 51
  • Dependent Packages: 23
  • Dependent Repositories: 37
  • Downloads: 3,704,885 Total
  • Rankings:
    • Stargazers count: 1.527%
    • Forks count: 2.64%
    • Dependent packages count: 2.904%
    • Average: 3.261%
    • Dependent repos count: 5.973%
spack.io: py-jaxlib

XLA library for Jax. jaxlib is the support library for JAX. While JAX itself is a pure Python package, jaxlib contains the binary (C/C++) parts of the library, including Python bindings, the XLA compiler, the PJRT runtime, and a handful of handwritten kernels.

  • Homepage: https://github.com/jax-ml/jax
  • Licenses: []
  • Latest release: 0.9.2 (published 12 days ago)
  • Last Synced: 2026-04-05T19:02:07.679Z (2 days ago)
  • Versions: 45
  • Dependent Packages: 6
  • Dependent Repositories: 0
  • Rankings:
    • Dependent repos count: 0.0%
    • Stargazers count: 0.403%
    • Forks count: 1.848%
    • Average: 5.308%
    • Dependent packages count: 18.98%
  • Maintainers (1)
anaconda.org: jax

JAX is Autograd and XLA, brought together for high-performance numerical computing and machine learning research. It provides composable transformations of Python+NumPy programs: differentiate, vectorize, parallelize, Just-In-Time compile to GPU/TPU, and more.

  • Homepage: https://github.com/jax-ml/jax
  • Licenses: Apache-2.0
  • Latest release: 0.7.2 (published 4 months ago)
  • Last Synced: 2026-03-25T18:07:32.230Z (13 days ago)
  • Versions: 6
  • Dependent Packages: 6
  • Dependent Repositories: 60
  • Downloads: 5,332 Total
  • Rankings:
    • Stargazers count: 4.016%
    • Dependent packages count: 7.065%
    • Forks count: 7.277%
    • Average: 10.311%
    • Dependent repos count: 22.886%
anaconda.org: jaxlib

JAX is a Python library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and large-scale machine learning.

  • Homepage: https://github.com/jax-ml/jax
  • Licenses: Apache-2.0
  • Latest release: 0.7.2 (published 4 months ago)
  • Last Synced: 2026-04-03T10:18:42.647Z (5 days ago)
  • Versions: 6
  • Dependent Packages: 5
  • Dependent Repositories: 37
  • Downloads: 6,717 Total
  • Rankings:
    • Stargazers count: 4.016%
    • Forks count: 7.277%
    • Dependent packages count: 8.756%
    • Average: 11.73%
    • Dependent repos count: 26.872%
pypi.org: jax-cuda13-plugin

JAX Plugin for NVIDIA GPUs

  • Homepage: https://github.com/jax-ml/jax
  • Documentation: https://jax-cuda13-plugin.readthedocs.io/
  • Licenses: Apache-2.0
  • Latest release: 0.9.2 (published 20 days ago)
  • Last Synced: 2026-04-05T19:02:07.258Z (2 days ago)
  • Versions: 10
  • Dependent Packages: 0
  • Dependent Repositories: 0
  • Downloads: 42,227 Last month
  • Rankings:
    • Dependent packages count: 8.827%
    • Average: 29.282%
    • Dependent repos count: 49.737%
  • Maintainers (3)
pypi.org: jax-cuda13-pjrt

JAX XLA PJRT Plugin for NVIDIA GPUs

  • Homepage: https://github.com/jax-ml/jax
  • Documentation: https://jax-cuda13-pjrt.readthedocs.io/
  • Licenses: Apache-2.0
  • Latest release: 0.9.2 (published 20 days ago)
  • Last Synced: 2026-04-05T19:02:21.943Z (2 days ago)
  • Versions: 10
  • Dependent Packages: 0
  • Dependent Repositories: 0
  • Downloads: 40,694 Last month
  • Rankings:
    • Dependent packages count: 8.827%
    • Average: 29.282%
    • Dependent repos count: 49.737%
  • Maintainers (3)
pypi.org: jax-rocm60-plugin

JAX Plugin for AMD GPUs (ROCm:unknown)

  • Homepage: https://github.com/jax-ml/jax
  • Documentation: https://jax-rocm60-plugin.readthedocs.io/
  • Licenses: Apache-2.0
  • Latest release: 0.5.0 (published about 1 year ago)
  • Last Synced: 2026-04-05T19:01:56.424Z (2 days ago)
  • Versions: 2
  • Dependent Packages: 0
  • Dependent Repositories: 0
  • Downloads: 2,135 Last month
  • Rankings:
    • Dependent packages count: 9.919%
    • Average: 32.879%
    • Dependent repos count: 55.838%
  • Maintainers (3)
pypi.org: jax-rocm60-pjrt

JAX XLA PJRT Plugin for AMD GPUs (ROCm:unknown)

  • Homepage: https://github.com/jax-ml/jax
  • Documentation: https://jax-rocm60-pjrt.readthedocs.io/
  • Licenses: Apache-2.0
  • Latest release: 0.5.0 (published 12 months ago)
  • Last Synced: 2026-04-05T19:01:58.207Z (2 days ago)
  • Versions: 2
  • Dependent Packages: 0
  • Dependent Repositories: 0
  • Downloads: 2,077 Last month
  • Rankings:
    • Dependent packages count: 9.919%
    • Average: 32.879%
    • Dependent repos count: 55.838%
  • Maintainers (3)
pypi.org: jax-cuda12-pjrt

JAX XLA PJRT Plugin for NVIDIA GPUs

  • Homepage: https://github.com/jax-ml/jax
  • Documentation: https://jax-cuda12-pjrt.readthedocs.io/
  • Licenses: Apache-2.0
  • Latest release: 0.9.2 (published 20 days ago)
  • Last Synced: 2026-04-05T19:02:15.451Z (2 days ago)
  • Versions: 33
  • Dependent Packages: 1
  • Dependent Repositories: 0
  • Downloads: 979,083 Last month
  • Rankings:
    • Dependent packages count: 10.156%
    • Average: 38.583%
    • Dependent repos count: 67.01%
  • Maintainers (1)
pypi.org: jax-cuda12-plugin

JAX Plugin for NVIDIA GPUs

  • Homepage: https://github.com/jax-ml/jax
  • Documentation: https://jax-cuda12-plugin.readthedocs.io/
  • Licenses: Apache-2.0
  • Latest release: 0.9.2 (published 20 days ago)
  • Last Synced: 2026-04-05T19:02:22.064Z (2 days ago)
  • Versions: 33
  • Dependent Packages: 1
  • Dependent Repositories: 0
  • Downloads: 962,118 Last month
  • Rankings:
    • Dependent packages count: 10.156%
    • Average: 38.583%
    • Dependent repos count: 67.01%
  • Maintainers (3)
nixpkgs-unstable: python314Packages.jax-cuda12-pjrt

JAX XLA PJRT Plugin for NVIDIA GPUs

nixpkgs-unstable: python313Packages.jax-cuda12-pjrt

JAX XLA PJRT Plugin for NVIDIA GPUs

nixpkgs-unstable: python314Packages.jax-cuda12-plugin

JAX Plugin for CUDA12

Dependencies

.github/workflows/ci-build.yaml actions
  • actions/cache 704facf57e6136b1bc63b828d79edcd491f0ee84 composite
  • actions/checkout b4ffde65f46336ab88eb53be808477a3936bae11 composite
  • actions/setup-python 0a5c61591373683505ea898e09a3ea4f39ef2b9c composite
  • pre-commit/action 646c83fcd040023954eafda54b4db0192ce70507 composite
  • styfle/cancel-workflow-action 01ce38bf961b4e243a6342cbade0dbc8ba3f0432 composite
.github/workflows/cloud-tpu-ci-nightly.yml actions
  • actions/checkout b4ffde65f46336ab88eb53be808477a3936bae11 composite
.github/workflows/jax-array-api.yml actions
  • actions/checkout b4ffde65f46336ab88eb53be808477a3936bae11 composite
  • actions/setup-python 0a5c61591373683505ea898e09a3ea4f39ef2b9c composite
.github/workflows/release-notification.yml actions
.github/workflows/wheel_win_x64.yml actions
  • actions/checkout b4ffde65f46336ab88eb53be808477a3936bae11 composite
  • actions/setup-python 0a5c61591373683505ea898e09a3ea4f39ef2b9c composite
  • actions/upload-artifact a8a3f3ad30e3422c9c7b888a15615d19a852ae32 composite
  • styfle/cancel-workflow-action 01ce38bf961b4e243a6342cbade0dbc8ba3f0432 composite
.github/workflows/windows_ci.yml actions
  • actions/checkout b4ffde65f46336ab88eb53be808477a3936bae11 composite
  • actions/setup-python 0a5c61591373683505ea898e09a3ea4f39ef2b9c composite
  • actions/upload-artifact a8a3f3ad30e3422c9c7b888a15615d19a852ae32 composite
  • styfle/cancel-workflow-action 01ce38bf961b4e243a6342cbade0dbc8ba3f0432 composite
build/collect-profile-requirements.txt pypi
  • protobuf *
  • tensorboard-plugin-profile *
  • tensorflow *
build/test-requirements.txt pypi
  • absl-py * test
  • build * test
  • cloudpickle * test
  • colorama >=0.4.4 test
  • flatbuffers * test
  • hypothesis * test
  • numpy >=1.22 test
  • pillow >=9.1.0 test
  • portpicker * test
  • pytest-xdist * test
  • rich * test
  • setuptools * test
  • wheel * test
docs/requirements.txt pypi
  • absl-py *
  • flatbuffers *
  • ipython >=8.8.0
  • jupyter-sphinx >=0.3.2
  • matplotlib *
  • myst-nb >=1.0.0
  • numpy *
  • pytest *
  • pytest-xdist *
  • scikit-learn *
  • sphinx >=6.0.0
  • sphinx-autodoc-typehints *
  • sphinx-book-theme >=1.0.1
  • sphinx-copybutton >=0.5.0
  • sphinx-design *
  • sphinx-remove-toctrees *
jax/experimental/jax2tf/examples/requirements.txt pypi
  • flax *
  • tensorflow_datasets *
  • tensorflow_hub *
jax_plugins/cuda/pyproject.toml pypi
jax_plugins/cuda/setup.py pypi
jaxlib/setup.py pypi
  • ml_dtypes >=0.2.0
  • numpy >=1.22
  • scipy >=1.11.1
  • scipy >=1.9
pyproject.toml pypi
setup.py pypi
  • Python *
  • Required *
  • importlib_metadata >=4.6
  • ml_dtypes >=0.2.0
  • numpy >=1.26.0
  • numpy >=1.22
  • numpy >=1.23.2
  • opt_einsum *
  • required *
  • scipy >=1.11.1
  • scipy >=1.9