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

jax

Score: 38.36542935088923

Last synced: about 16 hours ago
JSON representation

Repository metadata:

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

Host: GitHub
URL: https://github.com/jax-ml/jax
Owner: jax-ml
License: apache-2.0
Created: 2018-10-25T21:25:02.000Z (almost 8 years ago)
Default Branch: main
Last Pushed: 2026-08-19T13:02:20.000Z (6 days ago)
Last Synced: 2026-08-19T13:34:35.512Z (6 days ago)
Topics: jax
Language: Python
Homepage: https://docs.jax.dev
Size: 177 MB
Stars: 36,178
Watchers: 328
Forks: 3,738
Open Issues: 2,586
Metadata Files:
- Readme: README.md
- Changelog: CHANGELOG.md
- Contributing: CONTRIBUTING.md
- License: LICENSE
- Citation: CITATION.bib
- Authors: AUTHORS

Owner metadata:

Name: jax-ml
Login: jax-ml
Email:
Kind: organization
Description: miscellaneous libraries and projects relating to JAX
Website: github.com/google/jax
Location:
Twitter:
Company:
Icon url: https://avatars.githubusercontent.com/u/58486408?v=4
Repositories: 6
Last Synced at: 2023-03-06T21:12:23.854Z
Profile URL: https://github.com/jax-ml

Committers metadata

Last synced: 6 days ago

Total Commits: 28,528
Total Committers: 1,019
Avg Commits per committer: 27.996
Development Distribution Score (DDS): 0.882

Commits in past year: 6,701
Committers in past year: 298
Avg Commits per committer in past year: 22.487
Development Distribution Score (DDS) in past year: 0.88

Name	Email	Commits
Peter Hawkins	p**s@g**m	3367
Jake VanderPlas	j**p@g**m	2680
Matthew Johnson	m**j@g**m	2426
Yash Katariya	y**a@g**m	2278
jax authors	g**n@g**m	1804
George Necula	g**a@g**m	1252
Adam Paszke	a**e@g**m	1134
Sergei Lebedev	s**v@g**m	1069
jax authors	n**y@g**m	697
Roy Frostig	f**g@g**m	628
Benjamin Chetioui	b**i@g**m	505
Skye Wanderman-Milne	s**m@g**m	496
Sharad Vikram	s**v@g**m	471
Allan Renucci	a**i@g**m	385
Jake VanderPlas	v**s@g**m	358
Dan Foreman-Mackey	d**m@g**m	283
Tomás Longeri	t**i@g**m	242
Parker Schuh	p**s@g**m	239
Justin Fu	j**u@g**m	223
Daniel Suo	d**o@g**m	197
Dimitar (Mitko) Asenov	d**v@g**m	165
Chris Jones	c**j@g**m	159
Dougal	d**n@g**m	147
Stephan Hoyer	s**r@g**m	145
Jevin Jiang	j**g@g**m	142
Nitin Srinivasan	s**n@g**m	133
Greg Olechwierowicz	o**z@g**m	120
rajasekharporeddy	r**p@g**m	112
Vladimir Belitskiy	b**y@g**m	108
Lena Martens	l**s@g**m	107
and 989 more...

Issue and Pull Request metadata

Last synced: about 19 hours ago

Total issues: 1,336
Total pull requests: 10,397
Average time to close issues: 8 months
Average time to close pull requests: 15 days
Total issue authors: 709
Total pull request authors: 339
Average comments per issue: 2.44
Average comments per pull request: 0.25
Merged pull request: 5,800
Bot issues: 62
Bot pull requests: 7,513

Past year issues: 189
Past year pull requests: 1,398
Past year average time to close issues: 11 days
Past year average time to close pull requests: 21 days
Past year issue authors: 113
Past year pull request authors: 111
Past year average comments per issue: 1.28
Past year average comments per pull request: 0.28
Past year merged pull request: 602
Past year bot issues: 14
Past year bot pull requests: 1,085

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

Top Issue Authors

PhilipVinc (43)
carlosgmartin (42)
copybara-service[bot] (42)
ayaka14732 (28)
jakevdp (21)
github-actions[bot] (18)
patrick-kidger (17)
apiqwe (14)
hawkinsp (14)
sbodenstein (14)
gnecula (12)
vfdev-5 (12)
crusaderky (12)
mattjj (12)
emilyfertig (10)

Top Pull Request Authors

copybara-service[bot] (7,391)
jakevdp (566)
mattjj (239)
hawkinsp (205)
dfm (180)
gnecula (161)
dependabot[bot] (122)
vfdev-5 (73)
carlosgmartin (65)
rajasekharporeddy (50)
Ruturaj4 (47)
dougalm (42)
andportnoy (38)
froystig (32)
MichaelHudgins (32)

Top Issue Labels

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

Top Pull Request Labels

pull ready (1,968)
kokoro:force-run (727)
documentation (180)
dependencies (122)
python (82)
CI Optional GPU Presubmit (52)
github_actions (38)
CI Connection Halt - On Retry (20)
better_errors (14)
cla: yes (13)
windows:force-run (8)
AMD GPU (6)
needs info (4)
free threading (4)
CI (4)
skill issue (4)
enhancement (3)
bug (3)
CI Connection Halt - Always (2)
pallas (2)
NVIDIA GPU (2)
size/tiny (1)
Apple GPU (Metal) plugin (1)
build (1)
ok-to-test (1)
performance (1)
CI Connection Halt - On Error (1)
P3 (no schedule) (1)
ruby (1)
question (1)

Package metadata

Total packages: 25
Total downloads:
- pypi: 38,989,205 last-month
- conda: 5,411,840 total
Total docker downloads: 1,117,814,720
Total dependent packages: 1,385 (may contain duplicates)
Total dependent repositories: 10,632 (may contain duplicates)
Total versions: 713
Total maintainers: 24

nixpkgs-unstable: python313Packages.jax-cuda12-plugin

JAX Plugin for CUDA12

Homepage: https://github.com/jax-ml/jax/tree/main/jax_plugins/cuda
Documentation: https://github.com/NixOS/nixpkgs/blob/nixos-unstable/pkgs/development/python-modules/jax-cuda12-plugin/default.nix#L127
Licenses: Apache-2.0
Latest release: 0.9.2 (published 5 months ago)
Last Synced: 2026-04-02T19:03:02.873Z (5 months ago)
Versions: 2
Dependent Packages: 0
Dependent Repositories: 0
Rankings:
- Dependent repos count: 0.0%
- Dependent packages count: 0.0%
- Average: 0.044%
- Stargazers count: 0.077%
- Forks count: 0.098%
Maintainers (1)
- natsukium

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.11.1 (published 7 days ago)
Last Synced: 2026-08-23T15:32:33.844Z (1 day ago)
Versions: 195
Dependent Packages: 785
Dependent Repositories: 6,336
Downloads: 20,949,834 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)
- google_opensource
- jax-team
- mattjj
- phawkins
- gnecula
- Jake.Vanderplas
- jax-bot
- rf
- zhangqiaorjc

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.11.1 (published 7 days ago)
Last Synced: 2026-08-23T15:32:42.554Z (1 day ago)
Versions: 86
Dependent Packages: 511
Dependent Repositories: 4,102
Downloads: 15,580,488 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)
- google_opensource
- jax-team
- mattjj
- phawkins
- gnecula
- Jake.Vanderplas
- jax-bot
- rf
- zhangqiaorjc

conda-forge.org: jax

Homepage: https://github.com/jax-ml/jax
Licenses: Apache-2.0
Latest release: 0.3.25 (published almost 4 years ago)
Last Synced: 2026-03-25T18:08:14.239Z (5 months 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.10.2 (published 2 months ago)
Last Synced: 2026-08-23T15:32:35.481Z (1 day ago)
Versions: 58
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)
- adamjstewart

conda-forge.org: jaxlib

Homepage: http://github.com/jax-ml/jax
Licenses: Apache-2.0
Latest release: 0.3.25 (published almost 4 years ago)
Last Synced: 2026-03-25T18:08:13.142Z (5 months 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%

proxy.golang.org: github.com/jax-ml/jax

Homepage:
Status: removed
Documentation: https://pkg.go.dev/github.com/jax-ml/jax#section-documentation
Licenses: apache-2.0
Latest release: v0.0.0-20260811232613-cc77b3bbec99 (published 13 days ago)
Last Synced: 2026-08-23T15:35:29.619Z (1 day ago)
Versions: 1
Dependent Packages: 0
Dependent Repositories: 0
Rankings:
- Dependent packages count: 4.494%
- Average: 4.645%
- Dependent repos count: 4.796%

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.10.2 (published 2 months ago)
Last Synced: 2026-08-23T15:32:35.341Z (1 day ago)
Versions: 48
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 (2)
- adamjstewart
- jonas-eschle

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.9.2 (published 3 months ago)
Last Synced: 2026-05-13T20:05:50.389Z (3 months ago)
Versions: 7
Dependent Packages: 6
Dependent Repositories: 60
Downloads: 5,678 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.9.2 (published 4 months ago)
Last Synced: 2026-05-12T15:04:37.310Z (3 months ago)
Versions: 7
Dependent Packages: 5
Dependent Repositories: 37
Downloads: 7,400 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: winjax-cuda13-plugin

JAX Plugin for NVIDIA GPUs

Homepage: https://github.com/jax-ml/jax
Documentation: https://winjax-cuda13-plugin.readthedocs.io/
Licenses: Apache-2.0
Latest release: 0.11.0 (published 17 days ago)
Last Synced: 2026-08-23T15:32:44.760Z (1 day ago)
Versions: 1
Dependent Packages: 0
Dependent Repositories: 0
Downloads: 120 Last month
Rankings:
- Dependent packages count: 6.728%
- Average: 22.395%
- Dependent repos count: 38.063%
Maintainers (1)
- eterevsky

pypi.org: frxlib

XLA library for JAX

Homepage: https://github.com/jax-ml/jax
Documentation: https://frxlib.readthedocs.io/
Licenses: Apache-2.0
Latest release: 0.10.2 (published 13 days ago)
Last Synced: 2026-08-23T15:32:23.030Z (1 day ago)
Versions: 3
Dependent Packages: 0
Dependent Repositories: 0
Downloads: 437 Last month
Rankings:
- Dependent packages count: 6.805%
- Average: 22.652%
- Dependent repos count: 38.499%
Maintainers (1)
- fractalyze-contact

pypi.org: jax-oneapi-plugin

JAX Plugin for Intel GPUs

Homepage: https://github.com/jax-ml/jax
Documentation: https://jax-oneapi-plugin.readthedocs.io/
Licenses: Apache-2.0
Latest release: 0.0.1 (published 2 months ago)
Last Synced: 2026-08-23T15:32:36.830Z (1 day ago)
Versions: 2
Dependent Packages: 0
Dependent Repositories: 0
Downloads: 250 Last month
Rankings:
- Dependent packages count: 7.004%
- Average: 23.313%
- Dependent repos count: 39.621%
Maintainers (1)
- IntelAI

pypi.org: jax-oneapi-pjrt

JAX XLA PJRT Plugin for Intel GPUs

Homepage: https://github.com/jax-ml/jax
Documentation: https://jax-oneapi-pjrt.readthedocs.io/
Licenses: Apache-2.0
Latest release: 0.0.1 (published 2 months ago)
Last Synced: 2026-08-23T15:33:00.928Z (1 day ago)
Versions: 2
Dependent Packages: 0
Dependent Repositories: 0
Downloads: 53 Last month
Rankings:
- Dependent packages count: 7.004%
- Average: 25.414%
- Downloads: 29.618%
- Dependent repos count: 39.621%
Maintainers (1)
- IntelAI

pypi.org: jax-rocm7-pjrt

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

Homepage: https://github.com/jax-ml/jax
Documentation: https://jax-rocm7-pjrt.readthedocs.io/
Licenses: Apache-2.0
Latest release: 0.11.1 (published 7 days ago)
Last Synced: 2026-08-23T15:33:04.767Z (1 day ago)
Versions: 23
Dependent Packages: 0
Dependent Repositories: 0
Downloads: 13,294 Last month
Rankings:
- Dependent packages count: 8.493%
- Average: 28.249%
- Dependent repos count: 48.006%
Maintainers (7)

pypi.org: jax-rocm7-plugin

JAX Plugin for AMD GPUs (ROCm:unknown)

Homepage: https://github.com/jax-ml/jax
Documentation: https://jax-rocm7-plugin.readthedocs.io/
Licenses: Apache-2.0
Latest release: 0.11.1 (published 7 days ago)
Last Synced: 2026-08-23T15:32:36.269Z (1 day ago)
Versions: 23
Dependent Packages: 0
Dependent Repositories: 0
Downloads: 13,555 Last month
Rankings:
- Dependent packages count: 8.493%
- Average: 28.249%
- Dependent repos count: 48.006%
Maintainers (7)

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.11.1 (published 7 days ago)
Last Synced: 2026-08-23T15:32:41.570Z (1 day ago)
Versions: 15
Dependent Packages: 0
Dependent Repositories: 0
Downloads: 491,652 Last month
Rankings:
- Dependent packages count: 8.827%
- Average: 29.282%
- Dependent repos count: 49.737%
Maintainers (3)

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.11.1 (published 7 days ago)
Last Synced: 2026-08-23T15:32:37.060Z (1 day ago)
Versions: 15
Dependent Packages: 0
Dependent Repositories: 0
Downloads: 490,933 Last month
Rankings:
- Dependent packages count: 8.827%
- Average: 29.282%
- Dependent repos count: 49.737%
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 over 1 year ago)
Last Synced: 2026-08-23T15:32:58.972Z (1 day ago)
Versions: 2
Dependent Packages: 0
Dependent Repositories: 0
Downloads: 577 Last month
Rankings:
- Dependent packages count: 9.919%
- Average: 32.879%
- Dependent repos count: 55.838%
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 over 1 year ago)
Last Synced: 2026-08-23T15:32:31.215Z (1 day ago)
Versions: 2
Dependent Packages: 0
Dependent Repositories: 0
Downloads: 576 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.11.1 (published 7 days ago)
Last Synced: 2026-08-23T15:32:15.268Z (1 day ago)
Versions: 38
Dependent Packages: 1
Dependent Repositories: 0
Downloads: 722,910 Last month
Rankings:
- Dependent packages count: 10.156%
- Average: 38.583%
- Dependent repos count: 67.01%
Maintainers (3)

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.11.1 (published 7 days ago)
Last Synced: 2026-08-23T15:32:47.923Z (1 day ago)
Versions: 38
Dependent Packages: 1
Dependent Repositories: 0
Downloads: 724,526 Last month
Rankings:
- Dependent packages count: 10.156%
- Average: 38.583%
- Dependent repos count: 67.01%
Maintainers (3)

nixpkgs-unstable: python313Packages.jax-cuda12-pjrt

JAX XLA PJRT Plugin for NVIDIA GPUs

Homepage: https://github.com/jax-ml/jax/tree/main/jax_plugins/cuda
Documentation: https://github.com/NixOS/nixpkgs/blob/nixos-unstable/pkgs/development/python-modules/jax-cuda12-pjrt/default.nix#L95
Licenses: Apache-2.0
Latest release: 0.9.2 (published 5 months ago)
Last Synced: 2026-04-03T10:08:53.019Z (5 months ago)
Versions: 2
Dependent Packages: 0
Dependent Repositories: 0
Rankings:
- Dependent repos count: 0.0%
- Dependent packages count: 0.0%
- Average: 100%
Maintainers (1)
- natsukium

nixpkgs-unstable: python314Packages.jax-cuda12-plugin

JAX Plugin for CUDA12

Homepage: https://github.com/jax-ml/jax/tree/main/jax_plugins/cuda
Documentation: https://github.com/NixOS/nixpkgs/blob/nixos-unstable/pkgs/development/python-modules/jax-cuda12-plugin/default.nix#L127
Licenses: Apache-2.0
Latest release: 0.8.2 (published 7 months ago)
Last Synced: 2026-07-28T18:00:35.247Z (28 days ago)
Versions: 1
Dependent Packages: 0
Dependent Repositories: 0
Rankings:
- Dependent repos count: 0.0%
- Dependent packages count: 0.0%
- Average: 100%
Maintainers (1)
- natsukium

nixpkgs-unstable: python314Packages.jax-cuda12-pjrt

JAX XLA PJRT Plugin for NVIDIA GPUs

Homepage: https://github.com/jax-ml/jax/tree/main/jax_plugins/cuda
Documentation: https://github.com/NixOS/nixpkgs/blob/nixos-unstable/pkgs/development/python-modules/jax-cuda12-pjrt/default.nix#L95
Licenses: Apache-2.0
Latest release: 0.8.2 (published 7 months ago)
Last Synced: 2026-03-08T04:41:18.262Z (6 months ago)
Versions: 1
Dependent Packages: 0
Dependent Repositories: 0
Rankings:
- Dependent repos count: 0.0%
- Dependent packages count: 0.0%
- Average: 100%
Maintainers (1)
- natsukium

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

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

.github/actions/download-jax-cpu-wheels/action.yml actions

.github/workflows/bazel_cpu.yml actions

./.github/actions/download-jax-cpu-wheels * composite
./.github/actions/upload-test-artifacts * composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
google-ml-infra/actions/ci_connection 7f5ca0c263a81ed09ea276524c1b9192f1304e3c composite

.github/workflows/bazel_cpu_presubmit.yml actions

.github/workflows/build_rocm_artifacts.yml actions

actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
aws-actions/configure-aws-credentials e6de054238d6b7531b4efff3b6587d9aade6a06c composite
google-ml-infra/actions/ci_connection 7f5ca0c263a81ed09ea276524c1b9192f1304e3c composite

.github/workflows/k8s.yaml actions

actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
medyagh/setup-minikube cea33675329b799adccc9526aa5daccc26cd5052 composite

.github/workflows/lint.yml actions

actions/cache 55cc8345863c7cc4c66a329aec7e433d2d1c52a9 composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
google-ml-infra/actions/setup-uv-python a1817800cb84c752378772c2a02781cf8309a399 composite

.github/workflows/tsan.yaml actions

actions/cache/restore 55cc8345863c7cc4c66a329aec7e433d2d1c52a9 composite
actions/cache/save 0057852bfaa89a56745cba8c7296529d2fc39830 composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite

.github/workflows/upload_metadata.yml actions

actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite

.github/workflows/upstream-nightly.yml actions

./.github/actions/upload-test-artifacts * composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
actions/setup-python 5fda3b95a4ea91299a34e894583c3862153e4b97 composite
jayqi/failed-build-issue-action 1a893bbf43ef1c2a8705e2b115cd4f0fe3c5649b composite

.github/workflows/wheel_tests_continuous.yml actions

actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite

.github/workflows/wheel_tests_nightly_release.yml actions

MODULE.bazel bazel

abseil-cpp 20260526.0
apple_support 2.8.0
bazel_features 1.50.0
bazel_skylib 1.9.0
flatbuffers 25.2.10
googletest 1.17.0.bcr.2
grpc 1.78.0
platforms 1.1.0
protobuf 32.1
pybind11_bazel 3.0.0
rules_cc 0.2.20
rules_ml_toolchain *
rules_python 2.2.0
xla *
zlib 1.3.2

build/requirements.in pypi

build *
colorama >=0.4.4
etils *
jax *
jax-cuda12-pjrt *
jax-cuda12-plugin *
jax-cuda13-pjrt *
jax-cuda13-plugin *
jaxlib *
libtpu *
ml_dtypes >=0.5.3
numpy >=2.3.2
numpy *
opt-einsum *
scipy >=1.15.2
scipy >=1.15.0
setuptools *
tensorflow *
tensorstore *
wheel *
zstandard *

build/requirements_lock_3_13.txt pypi

absl-py ==2.5.0
astunparse ==1.6.3
auditwheel ==6.8.1
build ==1.5.0
certifi ==2026.7.22
charset-normalizer ==3.5.1
cloudpickle ==3.1.2
colorama ==0.4.6
contourpy ==1.3.3
cycler ==0.12.1
etils ==1.14.0
execnet ==2.1.2
filelock ==3.32.3
flatbuffers ==25.12.19
fonttools ==4.63.0
fsspec ==2026.7.0
gast ==0.7.0
google-pasta ==0.2.0
grpcio ==1.83.0
h5py ==3.14.0
hypothesis ==6.165.10
idna ==3.18
iniconfig ==2.3.0
jax ==0.11.1
jax-cuda12-pjrt ==0.11.1
jax-cuda12-plugin ==0.11.1
jax-cuda13-pjrt ==0.11.1
jax-cuda13-plugin ==0.11.1
jaxlib ==0.11.1
keras ==3.15.1
kiwisolver ==1.5.0
libclang ==18.1.1
libtpu ==0.0.46
markdown-it-py ==4.2.0
matplotlib ==3.11.1
mdurl ==0.1.2
ml-dtypes ==0.6.0
mpmath ==1.4.1
namex ==0.1.0
numpy ==2.1.3
numpy ==2.5.2
numpy-typing-compat ==20260602.2.1
numpy-typing-compat ==20260602.2.5
nvidia-cublas ==13.6.1.10
nvidia-cublas-cu12 ==12.9.2.10
nvidia-cuda-cccl ==13.3.3.4.1
nvidia-cuda-cccl-cu12 ==12.9.27
nvidia-cuda-crt ==13.3.73
nvidia-cuda-cupti ==13.3.75
nvidia-cuda-cupti-cu12 ==12.9.79
nvidia-cuda-nvcc ==13.3.73
nvidia-cuda-nvcc-cu12 ==12.9.86
nvidia-cuda-nvrtc ==13.3.33
nvidia-cuda-nvrtc-cu12 ==12.9.86
nvidia-cuda-runtime ==13.3.29
nvidia-cuda-runtime-cu12 ==12.9.79
nvidia-cudnn-cu12 ==9.24.0.43
nvidia-cudnn-cu13 ==9.24.0.43
nvidia-cufft ==12.3.0.29
nvidia-cufft-cu12 ==11.4.1.4
nvidia-cusolver ==12.2.6.9
nvidia-cusolver-cu12 ==11.7.5.82
nvidia-cusparse ==12.8.2.51
nvidia-cusparse-cu12 ==12.5.10.65
nvidia-nccl-cu12 ==2.31.2
nvidia-nccl-cu13 ==2.31.2
nvidia-nvjitlink ==13.3.33
nvidia-nvjitlink-cu12 ==12.9.86
nvidia-nvshmem-cu12 ==3.7.2
nvidia-nvshmem-cu13 ==3.7.2
nvidia-nvvm ==13.3.73
opt-einsum ==3.4.0
optree ==0.19.1
optype ==0.18.0
packaging ==26.3
pillow ==12.3.0
pluggy ==1.6.0
portpicker ==1.6.0
protobuf ==7.35.1
psutil ==7.2.2
pyelftools ==0.33
pygments ==2.21.0
pyparsing ==3.3.2
pyproject-hooks ==1.2.0
pytest ==9.1.1
pytest-xdist ==3.8.0
python-dateutil ==2.9.0.post0
requests ==2.34.2
rich ==15.0.0
scipy ==1.18.0
scipy-stubs ==1.18.0.1
setuptools ==84.0.0
six ==1.17.0
sortedcontainers ==2.4.0
tensorflow ==2.21.0
tensorstore ==0.1.85
termcolor ==3.3.0
typing-extensions ==4.16.0
urllib3 ==2.7.0
wheel ==0.48.0
wrapt ==2.3.0
zipp ==4.1.0
zstandard ==0.25.0

build/requirements_lock_3_15.txt pypi

absl-py ==2.5.0
auditwheel ==6.8.1
build ==1.5.0
cloudpickle ==3.1.2
colorama ==0.4.6
etils ==1.14.0
execnet ==2.1.2
filelock ==3.32.3
flatbuffers ==25.12.19
fsspec ==2026.7.0
hypothesis ==6.165.10
iniconfig ==2.3.0
jax ==0.11.1
jaxlib ==0.11.1
libtpu ==0.0.46
markdown-it-py ==4.2.0
mdurl ==0.1.2
ml-dtypes ==0.6.0
mpmath ==1.4.1
numpy ==2.5.2
numpy-typing-compat ==20260602.2.5
nvidia-cublas ==13.6.1.10
nvidia-cublas-cu12 ==12.9.2.10
nvidia-cuda-cccl ==13.3.3.4.1
nvidia-cuda-cccl-cu12 ==12.9.27
nvidia-cuda-crt ==13.3.73
nvidia-cuda-cupti ==13.3.75
nvidia-cuda-cupti-cu12 ==12.9.79
nvidia-cuda-nvcc ==13.3.73
nvidia-cuda-nvcc-cu12 ==12.9.86
nvidia-cuda-nvrtc ==13.3.33
nvidia-cuda-nvrtc-cu12 ==12.9.86
nvidia-cuda-runtime ==13.3.29
nvidia-cuda-runtime-cu12 ==12.9.79
nvidia-cudnn-cu12 ==9.24.0.43
nvidia-cudnn-cu13 ==9.24.0.43
nvidia-cufft ==12.3.0.29
nvidia-cufft-cu12 ==11.4.1.4
nvidia-cusolver ==12.2.6.9
nvidia-cusolver-cu12 ==11.7.5.82
nvidia-cusparse ==12.8.2.51
nvidia-cusparse-cu12 ==12.5.10.65
nvidia-nccl-cu12 ==2.31.2
nvidia-nccl-cu13 ==2.31.2
nvidia-nvjitlink ==13.3.33
nvidia-nvjitlink-cu12 ==12.9.86
nvidia-nvshmem-cu12 ==3.7.2
nvidia-nvshmem-cu13 ==3.7.2
nvidia-nvvm ==13.3.73
opt-einsum ==3.4.0
optype ==0.18.0
packaging ==26.3
pillow ==12.3.0
pluggy ==1.6.0
portpicker ==1.6.0
psutil ==7.2.2
pyelftools ==0.33
pygments ==2.21.0
pyproject-hooks ==1.2.0
pytest ==9.1.1
pytest-xdist ==3.8.0
rich ==15.0.0
scipy ==1.18.0
scipy-stubs ==1.18.0.1
setuptools ==84.0.0
sortedcontainers ==2.4.0
tensorstore ==0.1.85
typing-extensions ==4.16.0
wheel ==0.48.0
zipp ==4.1.0

.github/workflows/build_artifacts.yml actions

actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
google-ml-infra/actions/ci_connection 7f5ca0c263a81ed09ea276524c1b9192f1304e3c composite

.github/workflows/build_oneapi_artifacts.yml actions

actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
google-ml-infra/actions/ci_connection 7f5ca0c263a81ed09ea276524c1b9192f1304e3c composite

.github/workflows/numpy_nightly.yml actions

./.github/actions/upload-test-artifacts * composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
google-ml-infra/actions/ci_connection 7f5ca0c263a81ed09ea276524c1b9192f1304e3c composite

.github/workflows/oldest_supported_numpy.yml actions

./.github/actions/upload-test-artifacts * composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
google-ml-infra/actions/ci_connection 7f5ca0c263a81ed09ea276524c1b9192f1304e3c composite

.github/workflows/pytest_cpu.yml actions

./.github/actions/download-jax-cpu-wheels * composite
./.github/actions/upload-test-artifacts * composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
google-ml-infra/actions/ci_connection 7f5ca0c263a81ed09ea276524c1b9192f1304e3c composite

.github/workflows/pytest_cuda.yml actions

./.github/actions/download-jax-cuda-wheels * composite
./.github/actions/upload-test-artifacts * composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
google-ml-infra/actions/ci_connection 7f5ca0c263a81ed09ea276524c1b9192f1304e3c composite

.github/workflows/pytest_rocm.yml actions

./.github/actions/download-jax-rocm-wheels * composite
./.github/actions/upload-test-artifacts * composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
aws-actions/configure-aws-credentials e6de054238d6b7531b4efff3b6587d9aade6a06c composite
google-ml-infra/actions/ci_connection 7f5ca0c263a81ed09ea276524c1b9192f1304e3c composite

.github/workflows/pytest_tpu.yml actions

./.github/actions/download-jax-cpu-wheels * composite
./.github/actions/upload-test-artifacts * composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
google-ml-infra/actions/ci_connection 7f5ca0c263a81ed09ea276524c1b9192f1304e3c composite

.github/workflows/security.yml actions

actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
astral-sh/setup-uv c771a70e6277c0a99b617c7a806ffedaca235ff9 composite

.github/workflows/verify-squash.yml actions

actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite

build/nvidia-requirements.txt pypi

nvidia-cublas >=13.0.0.19
nvidia-cublas-cu12 >=12.1.3.1
nvidia-cuda-crt >=13.0.48
nvidia-cuda-cupti >=13.0.48
nvidia-cuda-cupti-cu12 >=12.1.105
nvidia-cuda-nvcc >=13.0.48
nvidia-cuda-nvcc-cu12 >=12.6.85
nvidia-cuda-nvrtc >=13.0.48
nvidia-cuda-nvrtc-cu12 >=12.1.55
nvidia-cuda-runtime >=13.0.48
nvidia-cuda-runtime-cu12 >=12.1.105
nvidia-cudnn-cu12 >=9.8,<10.0
nvidia-cudnn-cu13 >=9.12.0.46,<10.0
nvidia-cufft >=12.0.0.15
nvidia-cufft-cu12 >=11.0.2.54
nvidia-cusolver >=12.0.3.29
nvidia-cusolver-cu12 >=11.4.5.107
nvidia-cusparse >=12.6.2.49
nvidia-cusparse-cu12 >=12.1.0.106
nvidia-nccl-cu12 >=2.18.1
nvidia-nccl-cu13 >=2.27.7
nvidia-nvjitlink >=13.0.39
nvidia-nvjitlink-cu12 >=12.1.105
nvidia-nvshmem-cu12 >=3.2.5
nvidia-nvshmem-cu13 >=3.3.20
nvidia-nvvm >=13.0.48

build/requirements_lock_3_12.txt pypi

absl-py ==2.5.0
astunparse ==1.6.3
auditwheel ==6.8.1
build ==1.5.0
certifi ==2026.7.22
charset-normalizer ==3.5.1
cloudpickle ==3.1.2
colorama ==0.4.6
contourpy ==1.3.3
cycler ==0.12.1
etils ==1.14.0
execnet ==2.1.2
filelock ==3.32.3
flatbuffers ==25.12.19
fonttools ==4.63.0
fsspec ==2026.7.0
gast ==0.7.0
google-pasta ==0.2.0
grpcio ==1.83.0
h5py ==3.14.0
hypothesis ==6.165.10
idna ==3.18
iniconfig ==2.3.0
jax ==0.11.1
jax-cuda12-pjrt ==0.11.1
jax-cuda12-plugin ==0.11.1
jax-cuda13-pjrt ==0.11.1
jax-cuda13-plugin ==0.11.1
jaxlib ==0.11.1
keras ==3.15.1
kiwisolver ==1.5.0
libclang ==18.1.1
libtpu ==0.0.46
markdown-it-py ==4.2.0
matplotlib ==3.11.1
mdurl ==0.1.2
ml-dtypes ==0.6.0
mpmath ==1.4.1
namex ==0.1.0
numpy ==2.1.3
numpy ==2.5.2
numpy-typing-compat ==20260602.2.1
numpy-typing-compat ==20260602.2.5
nvidia-cublas ==13.6.1.10
nvidia-cublas-cu12 ==12.9.2.10
nvidia-cuda-cccl ==13.3.3.4.1
nvidia-cuda-cccl-cu12 ==12.9.27
nvidia-cuda-crt ==13.3.73
nvidia-cuda-cupti ==13.3.75
nvidia-cuda-cupti-cu12 ==12.9.79
nvidia-cuda-nvcc ==13.3.73
nvidia-cuda-nvcc-cu12 ==12.9.86
nvidia-cuda-nvrtc ==13.3.33
nvidia-cuda-nvrtc-cu12 ==12.9.86
nvidia-cuda-runtime ==13.3.29
nvidia-cuda-runtime-cu12 ==12.9.79
nvidia-cudnn-cu12 ==9.24.0.43
nvidia-cudnn-cu13 ==9.24.0.43
nvidia-cufft ==12.3.0.29
nvidia-cufft-cu12 ==11.4.1.4
nvidia-cusolver ==12.2.6.9
nvidia-cusolver-cu12 ==11.7.5.82
nvidia-cusparse ==12.8.2.51
nvidia-cusparse-cu12 ==12.5.10.65
nvidia-nccl-cu12 ==2.31.2
nvidia-nccl-cu13 ==2.31.2
nvidia-nvjitlink ==13.3.33
nvidia-nvjitlink-cu12 ==12.9.86
nvidia-nvshmem-cu12 ==3.7.2
nvidia-nvshmem-cu13 ==3.7.2
nvidia-nvvm ==13.3.73
opt-einsum ==3.4.0
optree ==0.19.1
optype ==0.18.0
packaging ==26.3
pillow ==12.3.0
pluggy ==1.6.0
portpicker ==1.6.0
protobuf ==7.35.1
psutil ==7.2.2
pyelftools ==0.33
pygments ==2.21.0
pyparsing ==3.3.2
pyproject-hooks ==1.2.0
pytest ==9.1.1
pytest-xdist ==3.8.0
python-dateutil ==2.9.0.post0
requests ==2.34.2
rich ==15.0.0
scipy ==1.18.0
scipy-stubs ==1.18.0.1
setuptools ==84.0.0
six ==1.17.0
sortedcontainers ==2.4.0
tensorflow ==2.21.0
tensorstore ==0.1.85
termcolor ==3.3.0
typing-extensions ==4.16.0
urllib3 ==2.7.0
wheel ==0.48.0
wrapt ==2.3.0
zipp ==4.1.0
zstandard ==0.25.0

build/requirements_lock_3_14.txt pypi

absl-py ==2.5.0
auditwheel ==6.8.1
build ==1.5.0
cloudpickle ==3.1.2
colorama ==0.4.6
contourpy ==1.3.3
cycler ==0.12.1
etils ==1.14.0
execnet ==2.1.2
filelock ==3.32.3
flatbuffers ==25.12.19
fonttools ==4.63.0
fsspec ==2026.7.0
hypothesis ==6.165.10
iniconfig ==2.3.0
jax ==0.11.1
jax-cuda12-pjrt ==0.11.1
jax-cuda12-plugin ==0.11.1
jax-cuda13-pjrt ==0.11.1
jax-cuda13-plugin ==0.11.1
jaxlib ==0.11.1
kiwisolver ==1.5.0
libtpu ==0.0.46
markdown-it-py ==4.2.0
matplotlib ==3.11.1
mdurl ==0.1.2
ml-dtypes ==0.6.0
mpmath ==1.4.1
numpy ==2.5.2
numpy-typing-compat ==20260602.2.5
nvidia-cublas ==13.6.1.10
nvidia-cublas-cu12 ==12.9.2.10
nvidia-cuda-cccl ==13.3.3.4.1
nvidia-cuda-cccl-cu12 ==12.9.27
nvidia-cuda-crt ==13.3.73
nvidia-cuda-cupti ==13.3.75
nvidia-cuda-cupti-cu12 ==12.9.79
nvidia-cuda-nvcc ==13.3.73
nvidia-cuda-nvcc-cu12 ==12.9.86
nvidia-cuda-nvrtc ==13.3.33
nvidia-cuda-nvrtc-cu12 ==12.9.86
nvidia-cuda-runtime ==13.3.29
nvidia-cuda-runtime-cu12 ==12.9.79
nvidia-cudnn-cu12 ==9.24.0.43
nvidia-cudnn-cu13 ==9.24.0.43
nvidia-cufft ==12.3.0.29
nvidia-cufft-cu12 ==11.4.1.4
nvidia-cusolver ==12.2.6.9
nvidia-cusolver-cu12 ==11.7.5.82
nvidia-cusparse ==12.8.2.51
nvidia-cusparse-cu12 ==12.5.10.65
nvidia-nccl-cu12 ==2.31.2
nvidia-nccl-cu13 ==2.31.2
nvidia-nvjitlink ==13.3.33
nvidia-nvjitlink-cu12 ==12.9.86
nvidia-nvshmem-cu12 ==3.7.2
nvidia-nvshmem-cu13 ==3.7.2
nvidia-nvvm ==13.3.73
opt-einsum ==3.4.0
optype ==0.18.0
packaging ==26.3
pillow ==12.3.0
pluggy ==1.6.0
portpicker ==1.6.0
psutil ==7.2.2
pyelftools ==0.33
pygments ==2.21.0
pyparsing ==3.3.2
pyproject-hooks ==1.2.0
pytest ==9.1.1
pytest-xdist ==3.8.0
python-dateutil ==2.9.0.post0
rich ==15.0.0
scipy ==1.18.0
scipy-stubs ==1.18.0.1
setuptools ==84.0.0
six ==1.17.0
sortedcontainers ==2.4.0
tensorstore ==0.1.85
typing-extensions ==4.16.0
wheel ==0.48.0
zipp ==4.1.0

.github/actions/download-jax-cuda-wheels/action.yml actions

.github/actions/download-jax-rocm-wheels/action.yml actions

.github/actions/upload-test-artifacts/action.yml actions

.github/workflows/asan.yaml actions

./jax/.github/actions/upload-test-artifacts * composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite

.github/workflows/bazel_cuda.yml actions

./.github/actions/download-jax-cuda-wheels * composite
./.github/actions/upload-test-artifacts * composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
google-ml-infra/actions/ci_connection 7f5ca0c263a81ed09ea276524c1b9192f1304e3c composite

.github/workflows/bazel_cuda_b200_mosaic_presubmit.yml actions

./.github/actions/upload-test-artifacts * composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
google-ml-infra/actions/ci_connection 9f8d7d8d9954430d0fef4a8cce2cb4e3844c200d composite

.github/workflows/bazel_cuda_h100_b200.yml actions

./.github/actions/upload-test-artifacts * composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
google-ml-infra/actions/ci_connection 7f5ca0c263a81ed09ea276524c1b9192f1304e3c composite
google-ml-infra/actions/ci_connection 9f8d7d8d9954430d0fef4a8cce2cb4e3844c200d composite

.github/workflows/bazel_cuda_presubmit.yml actions

.github/workflows/bazel_nobuild.yml actions

actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite

.github/workflows/bazel_oneapi_presubmit.yml actions

.github/workflows/bazel_rocm.yml actions

./.github/actions/download-jax-rocm-wheels * composite
./.github/actions/upload-test-artifacts * composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
aws-actions/configure-aws-credentials e6de054238d6b7531b4efff3b6587d9aade6a06c composite
google-ml-infra/actions/ci_connection 7f5ca0c263a81ed09ea276524c1b9192f1304e3c composite

.github/workflows/bazel_rocm_presubmit.yml actions

.github/workflows/bazel_test_tpu.yml actions

./.github/actions/download-jax-cpu-wheels * composite
./.github/actions/upload-test-artifacts * composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
google-ml-infra/actions/ci_connection 7f5ca0c263a81ed09ea276524c1b9192f1304e3c composite

.github/workflows/cloud-tpu-ci-presubmit.yml actions

.github/workflows/community_release_actions.yml actions

2428392/gh-truncate-string-action b3ff790d21cf42af3ca7579146eedb93c8fb0757 composite
tsickert/discord-webhook b217a69502f52803de774ded2b1ab7c282e99645 composite

.github/workflows/documentation.yml actions

actions/cache 55cc8345863c7cc4c66a329aec7e433d2d1c52a9 composite
actions/checkout 3d3c42e5aac5ba805825da76410c181273ba90b1 composite
google-ml-infra/actions/setup-uv-python a1817800cb84c752378772c2a02781cf8309a399 composite

build/requirements_lock_3_14_ft.txt pypi

absl-py ==2.5.0
auditwheel ==6.8.1
build ==1.5.0
cloudpickle ==3.1.2
colorama ==0.4.6
contourpy ==1.3.3
cycler ==0.12.1
etils ==1.14.0
execnet ==2.1.2
filelock ==3.32.3
flatbuffers ==25.12.19
fonttools ==4.63.0
fsspec ==2026.7.0
hypothesis ==6.165.10
iniconfig ==2.3.0
jax ==0.11.1
jax-cuda12-pjrt ==0.11.1
jax-cuda12-plugin ==0.11.1
jax-cuda13-pjrt ==0.11.1
jax-cuda13-plugin ==0.11.1
jaxlib ==0.11.1
kiwisolver ==1.5.0
libtpu ==0.0.46
markdown-it-py ==4.2.0
matplotlib ==3.11.1
mdurl ==0.1.2
ml-dtypes ==0.6.0
mpmath ==1.4.1
numpy ==2.5.2
numpy-typing-compat ==20260602.2.5
nvidia-cublas ==13.6.1.10
nvidia-cublas-cu12 ==12.9.2.10
nvidia-cuda-cccl ==13.3.3.4.1
nvidia-cuda-cccl-cu12 ==12.9.27
nvidia-cuda-crt ==13.3.73
nvidia-cuda-cupti ==13.3.75
nvidia-cuda-cupti-cu12 ==12.9.79
nvidia-cuda-nvcc ==13.3.73
nvidia-cuda-nvcc-cu12 ==12.9.86
nvidia-cuda-nvrtc ==13.3.33
nvidia-cuda-nvrtc-cu12 ==12.9.86
nvidia-cuda-runtime ==13.3.29
nvidia-cuda-runtime-cu12 ==12.9.79
nvidia-cudnn-cu12 ==9.24.0.43
nvidia-cudnn-cu13 ==9.24.0.43
nvidia-cufft ==12.3.0.29
nvidia-cufft-cu12 ==11.4.1.4
nvidia-cusolver ==12.2.6.9
nvidia-cusolver-cu12 ==11.7.5.82
nvidia-cusparse ==12.8.2.51
nvidia-cusparse-cu12 ==12.5.10.65
nvidia-nccl-cu12 ==2.31.2
nvidia-nccl-cu13 ==2.31.2
nvidia-nvjitlink ==13.3.33
nvidia-nvjitlink-cu12 ==12.9.86
nvidia-nvshmem-cu12 ==3.7.2
nvidia-nvshmem-cu13 ==3.7.2
nvidia-nvvm ==13.3.73
opt-einsum ==3.4.0
optype ==0.18.0
packaging ==26.3
pillow ==12.3.0
pluggy ==1.6.0
portpicker ==1.6.0
psutil ==7.2.2
pyelftools ==0.33
pygments ==2.21.0
pyparsing ==3.3.2
pyproject-hooks ==1.2.0
pytest ==9.1.1
pytest-xdist ==3.8.0
python-dateutil ==2.9.0.post0
rich ==15.0.0
scipy ==1.18.0
scipy-stubs ==1.18.0.1
setuptools ==84.0.0
six ==1.17.0
sortedcontainers ==2.4.0
tensorstore ==0.1.85
typing-extensions ==4.16.0
wheel ==0.48.0
zipp ==4.1.0

build/requirements_lock_3_15_ft.txt pypi

absl-py ==2.5.0
auditwheel ==6.8.1
build ==1.5.0
cloudpickle ==3.1.2
colorama ==0.4.6
etils ==1.14.0
execnet ==2.1.2
filelock ==3.32.3
flatbuffers ==25.12.19
fsspec ==2026.7.0
hypothesis ==6.165.10
iniconfig ==2.3.0
jax ==0.11.1
jaxlib ==0.11.1
libtpu ==0.0.46
markdown-it-py ==4.2.0
mdurl ==0.1.2
ml-dtypes ==0.6.0
mpmath ==1.4.1
numpy ==2.5.2
numpy-typing-compat ==20260602.2.5
nvidia-cublas ==13.6.1.10
nvidia-cublas-cu12 ==12.9.2.10
nvidia-cuda-cccl ==13.3.3.4.1
nvidia-cuda-cccl-cu12 ==12.9.27
nvidia-cuda-crt ==13.3.73
nvidia-cuda-cupti ==13.3.75
nvidia-cuda-cupti-cu12 ==12.9.79
nvidia-cuda-nvcc ==13.3.73
nvidia-cuda-nvcc-cu12 ==12.9.86
nvidia-cuda-nvrtc ==13.3.33
nvidia-cuda-nvrtc-cu12 ==12.9.86
nvidia-cuda-runtime ==13.3.29
nvidia-cuda-runtime-cu12 ==12.9.79
nvidia-cudnn-cu12 ==9.24.0.43
nvidia-cudnn-cu13 ==9.24.0.43
nvidia-cufft ==12.3.0.29
nvidia-cufft-cu12 ==11.4.1.4
nvidia-cusolver ==12.2.6.9
nvidia-cusolver-cu12 ==11.7.5.82
nvidia-cusparse ==12.8.2.51
nvidia-cusparse-cu12 ==12.5.10.65
nvidia-nccl-cu12 ==2.31.2
nvidia-nccl-cu13 ==2.31.2
nvidia-nvjitlink ==13.3.33
nvidia-nvjitlink-cu12 ==12.9.86
nvidia-nvshmem-cu12 ==3.7.2
nvidia-nvshmem-cu13 ==3.7.2
nvidia-nvvm ==13.3.73
opt-einsum ==3.4.0
optype ==0.18.0
packaging ==26.3
pillow ==12.3.0
pluggy ==1.6.0
portpicker ==1.6.0
psutil ==7.2.2
pyelftools ==0.33
pygments ==2.21.0
pyproject-hooks ==1.2.0
pytest ==9.1.1
pytest-xdist ==3.8.0
rich ==15.0.0
scipy ==1.18.0
scipy-stubs ==1.18.0.1
setuptools ==84.0.0
sortedcontainers ==2.4.0
tensorstore ==0.1.85
typing-extensions ==4.16.0
wheel ==0.48.0
zipp ==4.1.0

examples/ffi/pyproject.toml pypi

jax *

jax_plugins/oneapi/pyproject.toml pypi

jax_plugins/oneapi/setup.py pypi

jax_plugins/rocm/pyproject.toml pypi

jax_plugins/rocm/setup.py pypi

jaxlib/mosaic/gpu/wheel/setup.py pypi