nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-12-14 01:33:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
c0e8b179c5
nixpkgs/pkgs/development/python-modules/jax
History
Alexander Tsvyashchenko be52722509
python3Packages.jaxlib: refactor to support Nix-based builds (#151909) 
* python3Packages.jaxlib: rename to `jaxlib-bin`

Refactoring `jaxlib` to have a similar structure to `tensorflow` with the 'bin' and 'build' options.

* python3Packages.jaxlib: init the 'build' variant at 0.1.75

Similar to `tensorflow-build`, now there's an option to build `jaxlib` using Nix-provided environment and dependencies.

* python3Packages.jax: 0.2.24 -> 0.2.26

* Addressed review comments.

* Fixed `cudaSupport` missing property on some arches.

* Unified the versions of CUDA-related packages with TF.

Co-authored-by: Samuel Ainsworth <skainsworth@gmail.com>
2021-12-27 16:19:10 -08:00
..
cache-fix.patch python3Packages.jaxlib: refactor to support Nix-based builds (#151909) 2021-12-27 16:19:10 -08:00
default.nix python3Packages.jaxlib: refactor to support Nix-based builds (#151909) 2021-12-27 16:19:10 -08:00