5 commits

Author	SHA1	Message	Date
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	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