python313Packages.jax 0.11.2 maintained
Source-built JAX frontend: differentiate, compile, and transform Numpy code
- Name
python3.13-jax-0.11.2- Package set
- python313Packages
- Homepage
- https://github.com/jax-ml/jax
- License
- Apache-2.0
- Source
- pkgs/development/python-modules/jax/default.nix:175
- Derivation
/nix/store/rah50i343n3bklxwqp661mfb2gsqg3d1-python3.13-jax-0.11.2.drv
Maintainers 6
| Handle | Name | Contact | Via |
|---|---|---|---|
| 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
- cuda-maintainers: Maintain CUDA-enabled packages
Dependencies 28
| Package | As | Maintainers | Used by (transitive) | Status |
|---|---|---|---|---|
| python313 3.13.15 | propagatedBuildInputs | 13,660 | team-only | |
| python313Packages.jaxlib 0.11.2 | propagatedBuildInputs | 81 | single | |
| python313Packages.ml-dtypes 0.6.0 | propagatedBuildInputs | 138 | maintained | |
| python313Packages.numpy 2.5.2 | propagatedBuildInputs | 5,137 | single | |
| python313Packages.opt-einsum 3.4.0 | propagatedBuildInputs | 287 | single | |
| python313Packages.scipy 1.18.1 | propagatedBuildInputs | 1,338 | single | |
| ensureNewerSourcesForZipFilesHook 26.05pre-git | nativeBuildInputs | 74,596 | unmaintained setup hook | |
| python313Packages.absl-py 2.5.0 | nativeBuildInputs | 173 | unmaintained | |
| python313Packages.cloudpickle 3.1.2 | nativeBuildInputs | 3,929 | unmaintained | |
| python313Packages.flatbuffers 25.12.19 | nativeBuildInputs | 105 | team-only | |
| python313Packages.hypothesis 6.156.1 | nativeBuildInputs | 6,765 | single | |
| python313Packages.matplotlib 3.11.1 | nativeBuildInputs | 4,079 | maintained | |
| python313Packages.pypaBuildHook 26.05pre-git | nativeBuildInputs | 11,598 | unmaintained setup hook | |
| python313Packages.pypaInstallHook 26.05pre-git | nativeBuildInputs | 11,625 | unmaintained setup hook | |
| python313Packages.pytest-xdist 3.8.0 | nativeBuildInputs | 7,922 | single | |
| python313Packages.pytestCheckHook 26.05pre-git | nativeBuildInputs | 10,182 | unmaintained setup hook | |
| python313Packages.pythonCatchConflictsHook 26.05pre-git | nativeBuildInputs | 11,609 | unmaintained setup hook | |
| python313Packages.pythonImportsCheckHook 26.05pre-git | nativeBuildInputs | 11,642 | unmaintained setup hook | |
| python313Packages.pythonMetadataCheckHook 26.05pre-git | nativeBuildInputs | 11,597 | single setup hook | |
| python313Packages.pythonNamespacesHook 26.05pre-git | nativeBuildInputs | 11,634 | 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 | |
| python313Packages.pythonRemoveTestsDirHook 26.05pre-git | nativeBuildInputs | 11,634 | unmaintained setup hook | |
| python313Packages.pythonRuntimeDepsCheckHook 26.05pre-git | nativeBuildInputs | 11,611 | unmaintained setup hook | |
| python313Packages.setuptools 83.0.0 | nativeBuildInputs | 11,500 | team-only | |
| python313Packages.wrapPython 26.05pre-git | nativeBuildInputs | 11,634 | unmaintained setup hook | |
| bashNonInteractive 5.3p15 | other input | 102,441 | maintained | |
| stdenv 26.05pre-git | other input | 90,330 | team-only |
Used by 45 directly, 80 transitively
| Package | As | Maintainers | Used by (transitive) | Status |
|---|---|---|---|---|
| python313Packages.chex 0.1.92 | propagatedBuildInputs | 51 | single | |
| python313Packages.etils 1.14.0 | nativeBuildInputs | 48 | single | |
| python313Packages.optax 0.2.8 | propagatedBuildInputs | 47 | single | |
| python313Packages.orbax-checkpoint 0.12.6 | propagatedBuildInputs | 39 | single | |
| python313Packages.keras 3.15.1 | nativeBuildInputs | 37 | single | |
| python313Packages.treescope 0.1.10 | nativeBuildInputs | 26 | single | |
| python313Packages.flax 0.12.10 | propagatedBuildInputs | 25 | single | |
| python313Packages.array-api-compat 1.15 | nativeBuildInputs | 14 | single | |
| python313Packages.equinox 0.13.8 | propagatedBuildInputs | 12 | single | |
| python313Packages.jmp 0.0.4 | propagatedBuildInputs | 11 | single | |
| python313Packages.gymnasium 1.3.0 | nativeBuildInputs | 10 | single | |
| python313Packages.numpyro 0.22.0 | propagatedBuildInputs | 7 | single | |
| python313Packages.ale-py 0.12.1 | nativeBuildInputs | 4 | single | |
| python313Packages.outlines 1.2.12 | nativeBuildInputs | 4 | single | |
| python313Packages.pyvers 0.2.3 | nativeBuildInputs | 4 | single | |
| python313Packages.botorch 0.18.1 | nativeBuildInputs | 3 | single | |
| python313Packages.pytensor 3.3.3 | nativeBuildInputs | 3 | single | |
| python313Packages.jax-tap 0.3.1 | propagatedBuildInputs | 2 | single | |
| python313Packages.jaxopt 0.8.5 | propagatedBuildInputs | 2 | single | |
| python313Packages.mujoco-mjx 3.14.0 | propagatedBuildInputs | 2 | single | |
| python313Packages.paramax 0.0.5 | propagatedBuildInputs | 2 | single | |
| python313Packages.tensorflow-datasets 4.9.10 | nativeBuildInputs | 2 | single | |
| python313Packages.blackjax 1.7.1 | propagatedBuildInputs | 1 | single | |
| python313Packages.brax 0.14.2 | propagatedBuildInputs | 1 | single | |
| python313Packages.distrax 0.1.9 | propagatedBuildInputs | 1 | single | |
| python313Packages.flowjax 18.0.0 | propagatedBuildInputs | 1 | single | |
| python313Packages.lineax 0.1.1 | propagatedBuildInputs | 1 | single | |
| python313Packages.augmax 0.3.3 | propagatedBuildInputs | 0 | single | |
| python313Packages.ax-platform 1.3.1 | nativeBuildInputs | 0 | single | |
| python313Packages.clu 0.0.12 | propagatedBuildInputs | 0 | single | |
| python313Packages.daft 0.7.25 | nativeBuildInputs | 0 | single | |
| python313Packages.evosax 0.3.1 | propagatedBuildInputs | 0 | single | |
| python313Packages.flowmc 0.4.5 | propagatedBuildInputs | 0 | single | |
| python313Packages.minari 0.5.4 | nativeBuildInputs | 0 | single | |
| python313Packages.nifty8 8.5.7 | propagatedBuildInputs | 0 | single | |
| python313Packages.nutpie 0.16.11 | nativeBuildInputs | 0 | single | |
| python313Packages.optimistix 0.0.11 | propagatedBuildInputs | 0 | single | |
| python313Packages.orbax-export 0.0.8 | propagatedBuildInputs | 0 | single | |
| python313Packages.oryx 0.2.9 | propagatedBuildInputs | 0 | single broken | |
| python313Packages.pymanopt 2.2.1 | nativeBuildInputs | 0 | single | |
| python313Packages.rlax 0.1.9 | propagatedBuildInputs | 0 | single | |
| python313Packages.skrl 2.1.0 | nativeBuildInputs | 0 | single | |
| python313Packages.transformer-engine 2.19 | propagatedBuildInputs | 0 | maintained broken | |
| python313Packages.transformer-engine-jax 2.19 | propagatedBuildInputs | 0 | maintained broken | |
| python313Packages.waymax 0-unstable-2025-05-30 | propagatedBuildInputs | 0 | single unfree |