b12f/nixpkgs
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity
365951 commits 2 branches 0 tags 3.3 GiB
Commit graph

12 commits

Author SHA1 Message Date
Samuel Ainsworth f9588d6966 python3Packages.{jaxlibWithCuda, jaxlib-bin}: add ptxas to $out/bin 2022-03-17 21:08:55 +00:00
Samuel Ainsworth 2ddeac059b python3Packages.jaxlib-bin: add support for python 3.10 and cudnn >=8.2 2022-03-04 03:04:12 +00:00
Samuel Ainsworth ab9d59200c python3Packages.jaxlib-bin: comment explaining cudatoolkit dependency 2022-02-18 21:27:28 +00:00
Samuel Ainsworth e663c60265 python3Packages.jaxlib: 0.1.75 -> 0.3.0 2022-02-14 21:41:10 +00:00
Samuel Ainsworth d81549cd38 python3Packages.jaxlib-bin: 0.1.75 -> 0.3.0 2022-02-14 21:40:45 +00:00
Samuel Ainsworth d6e2771695 python3Packages.jaxlib-bin: 0.1.71 -> 0.1.75 2022-02-05 00:55:01 +00:00
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
Markus S. Wamser 9b001407a1 pkgs.development.python-modules: remove unused args 2021-11-13 13:43:29 +01:00
Samuel Ainsworth 8f06bd48f2 python3Packages.jaxlib: add cudatoolkit_11 to rpaths 2021-09-04 21:42:04 +00:00
Samuel Ainsworth 3ea5dbdd72 python3Packages.jaxlib: add cudatoolkit_11 in propagatedBuildInputs 2021-09-04 20:58:15 +00:00
Samuel Ainsworth 0ff986154e python3Packages.jaxlib: add CUDA support 2021-09-04 05:47:09 +00:00
Samuel Ainsworth 1f8686373a python3Packages.jaxlib: init at 0.1.70 2021-08-22 20:37:42 +00:00