x86_64-linux · 151fa4e8ddfd · generated 2026-10-07T10:25:31Z

python314Packages.jax 0.11.2 maintained

Source-built JAX frontend: differentiate, compile, and transform Numpy code

Name
python3.14-jax-0.11.2
Package set
python314Packages
Homepage
https://github.com/jax-ml/jax
License
Apache-2.0
Source
pkgs/development/python-modules/jax/default.nix:175
Derivation
/nix/store/dscmb80ys56g4lm6b8s5qc92m175yrki-python3.14-jax-0.11.2.drv
Repology
not reported as outdated · look up (as of 2026-10-07T10:23:38Z)

Maintainers 6

HandleNameContactVia
GaetanLepage Gaetan Lepage GitHub · gaetan@glepage.com listed on package, team cuda-maintainers
samuela Samuel Ainsworth GitHub · skainsworth@gmail.com listed on package, team cuda-maintainers
ConnorBaker Connor Baker GitHub · ConnorBaker01@gmail.com · @connorbaker:matrix.org team cuda-maintainers
ethancedwards8 Ethan Carter Edwards GitHub · ethan@ethancedwards.com · @ethancedwards8:matrix.org team cuda-maintainers
SomeoneSerge Else Someone GitHub · else+nixpkgs@someonex.net · @ss:someonex.net team cuda-maintainers
YorikSar Yuriy Taraday GitHub · yorik.sar@gmail.com · @yorik.sar:matrix.org team cuda-maintainers

Teams 1

Dependencies 28

PackageAsMaintainersUsed by (transitive)Status
python3 3.14.7 propagatedBuildInputs 72,537 team-only outdated
python314Packages.jaxlib 0.11.2 propagatedBuildInputs 37 single
python314Packages.ml-dtypes 0.6.0 propagatedBuildInputs 207 maintained
python314Packages.numpy 2.5.2 propagatedBuildInputs 11,439 single outdated
python314Packages.opt-einsum 3.4.0 propagatedBuildInputs 271 single
python314Packages.scipy 1.18.1 propagatedBuildInputs 1,775 single
ensureNewerSourcesForZipFilesHook 26.05pre-git nativeBuildInputs
74,596 unmaintained setup hook
python313Packages.pythonOutputDistHook 26.05pre-git nativeBuildInputs
74,564 unmaintained setup hook
python313Packages.pythonRemoveBinBytecodeHook 26.05pre-git nativeBuildInputs
74,584 unmaintained setup hook
python314Packages.absl-py 2.5.0 nativeBuildInputs
282 unmaintained
python314Packages.cloudpickle 3.1.2 nativeBuildInputs
6,078 unmaintained
python314Packages.flatbuffers 25.12.19 nativeBuildInputs 63 team-only
python314Packages.hypothesis 6.156.1 nativeBuildInputs 13,458 single outdated
python314Packages.matplotlib 3.11.1 nativeBuildInputs 6,909 maintained outdated
python314Packages.pypaBuildHook 26.05pre-git nativeBuildInputs
71,473 unmaintained setup hook
python314Packages.pypaInstallHook 26.05pre-git nativeBuildInputs
71,694 unmaintained setup hook
python314Packages.pytest-xdist 3.8.0 nativeBuildInputs 44,257 single
python314Packages.pytestCheckHook 26.05pre-git nativeBuildInputs
70,075 unmaintained setup hook
python314Packages.pythonCatchConflictsHook 26.05pre-git nativeBuildInputs
71,690 unmaintained setup hook
python314Packages.pythonImportsCheckHook 26.05pre-git nativeBuildInputs
71,715 unmaintained setup hook
python314Packages.pythonMetadataCheckHook 26.05pre-git nativeBuildInputs 71,460 single setup hook
python314Packages.pythonNamespacesHook 26.05pre-git nativeBuildInputs
71,713 unmaintained setup hook
python314Packages.pythonRemoveTestsDirHook 26.05pre-git nativeBuildInputs
71,713 unmaintained setup hook
python314Packages.pythonRuntimeDepsCheckHook 26.05pre-git nativeBuildInputs
71,662 unmaintained setup hook
python314Packages.setuptools 83.0.0 nativeBuildInputs 71,381 team-only outdated
python314Packages.wrapPython 26.05pre-git nativeBuildInputs
71,719 unmaintained setup hook
bashNonInteractive 5.3p15 other input 102,441 maintained outdated
stdenv 26.05pre-git other input
90,330 team-only

Used by 21 directly, 36 transitively

PackageAsMaintainersUsed by (transitive)Status
python314Packages.chex 0.1.92 propagatedBuildInputs 11 single
python314Packages.etils 1.14.0 nativeBuildInputs 8 single
python314Packages.optax 0.2.8 propagatedBuildInputs 8 single
python314Packages.array-api-compat 1.15 nativeBuildInputs 4 single
python314Packages.equinox 0.13.8 propagatedBuildInputs 4 single
python314Packages.pyvers 0.2.3 nativeBuildInputs 3 single
python314Packages.jax-tap 0.3.1 propagatedBuildInputs 1 single
python314Packages.jmp 0.0.4 propagatedBuildInputs 1 single
python314Packages.lineax 0.1.1 propagatedBuildInputs 1 single
python314Packages.pytensor 3.3.3 nativeBuildInputs 1 single
python314Packages.augmax 0.3.3 propagatedBuildInputs 0 single outdated
python314Packages.blackjax 1.7.1 propagatedBuildInputs 0 single
python314Packages.daft 0.7.25 nativeBuildInputs 0 single outdated
python314Packages.flowmc 0.4.5 propagatedBuildInputs 0 single outdated
python314Packages.jaxopt 0.8.5 propagatedBuildInputs 0 single
python314Packages.mujoco-mjx 3.14.0 propagatedBuildInputs 0 single
python314Packages.nifty8 8.5.7 propagatedBuildInputs 0 single
python314Packages.optimistix 0.0.11 propagatedBuildInputs 0 single outdated
python314Packages.orbax-checkpoint 0.12.6 propagatedBuildInputs 0 single
python314Packages.paramax 0.0.5 propagatedBuildInputs 0 single
python314Packages.treescope 0.1.10 nativeBuildInputs 0 single