nixsearch
nixsearch / pkg / python312Packages.jaxlibWithCuda / 0.4.28

python312Packages.jaxlibWithCuda 0.4.28

Source-built JAX backend. JAX is Autograd and XLA, brought together for high-performance machine learning research
Linux last changed outputs: out (Linux only), dist (Linux only)

Pin this version

$ devbox add python312Packages.jaxlibWithCuda@0.4.28
systemsx86_64-linux commitde1864217bfa9b5845f465e771e0ecb48b30e02d 
$ nix shell github:NixOS/nixpkgs/de1864217bfa9b5845f465e771e0ecb48b30e02d#python312Packages.jaxlibWithCuda

Per system

SystemAttributeCommitDateOutputsFlags
x86_64-linux
Linux · x86-64		 python312Packages.jaxlibWithCuda de1864217bfa
out
/nix/store/dd3p8p1qm642nvfhjjl1wfgad5l8sp9p-python3.12-jaxlib-0.4.28
dist (not default)
/nix/store/jsbfri24bpvi4zpch0vbms164v0gw15q-python3.12-jaxlib-0.4.28-dist
all 3 releases older: 0.4.24 →

JSON: /v2/pkg?name=python312Packages.jaxlibWithCuda · /v2/resolve?name=python312Packages.jaxlibWithCuda&version=0.4.28