nixos-unstable · x86_64-linux · 151fa4e8ddfd · generated 2026-10-07T01:01:54Z

python313Packages.jmp 0.0.4 single

This library implements support for mixed precision training in JAX

Name
python3.13-jmp-0.0.4
Package set
python313Packages
Homepage
https://github.com/deepmind/jmp
License
Apache-2.0
Source
pkgs/development/python-modules/jmp/default.nix:49
Derivation
/nix/store/cm3353w3dw2kay66jkd1cv5a8fx5gfpf-python3.13-jmp-0.0.4.drv

Maintainers 1

HandleNameContactVia
ndl Alexander Tsvyashchenko GitHub · ndl@endl.ch listed on package

Dependencies 20

PackageAsMaintainersUsed by (transitive)Status
python313 3.13.15 propagatedBuildInputs
mweinelt
 13,660 team-only
python313Packages.absl-py 2.5.0 propagatedBuildInputs
173 unmaintained
python313Packages.jax 0.11.2 propagatedBuildInputs
ConnorBaker, ethancedwards8, GaetanLepage, samuela, SomeoneSerge, YorikSar
 80 maintained
python313Packages.numpy 2.5.2 propagatedBuildInputs
doronbehar
 5,137 single
ensureNewerSourcesForZipFilesHook 26.05pre-git nativeBuildInputs
74,596 unmaintained setup hook
python313Packages.pypaBuildHook 26.05pre-git nativeBuildInputs
11,598 unmaintained setup hook
python313Packages.pypaInstallHook 26.05pre-git nativeBuildInputs
11,625 unmaintained setup hook
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
dotlambda
 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
mweinelt
 11,500 team-only
python313Packages.wrapPython 26.05pre-git nativeBuildInputs
11,634 unmaintained setup hook
bashNonInteractive 5.3p15 other input
andir, balsoft, infinisil, pyrox0
 102,441 maintained
stdenv 26.05pre-git other input
90,330 team-only

Used by 1 directly, 11 transitively

PackageAsMaintainersUsed by (transitive)Status
python313Packages.dm-haiku 0.0.17 propagatedBuildInputs
ndl
 10 single