Commit graph

14 commits

Author SHA1 Message Date
R. Ryantm 011f9bfd39 python310Packages.jax: 0.3.3 -> 0.3.4 2022-03-19 03:15:53 +00:00
nixpkgs-upkeep-bot a05a02a88a python3Packages.jax: 0.3.1 -> 0.3.3 2022-03-18 00:37:10 +00:00
Samuel Ainsworth 069b742520 python3Packages.jax: fix MKL-enabled build on Intel CPUs
Intel doesn't know how to build a CPU it turns out. Don't use them for floating point arithmetic, even with their own MKL!
2022-02-27 08:06:53 +00:00
Samuel Ainsworth 33984cd89c python3Packages.jax: support MKL BLAS/LAPACK implementations 2022-02-26 07:33:28 +01:00
nixpkgs-upkeep-bot cef32aee43 python3Packages.jax: 0.3.0 -> 0.3.1 2022-02-19 00:42:06 +00:00
Samuel Ainsworth e9015499e0 python3Packages.jax: 0.2.28 -> 0.3.0 2022-02-14 21:40:19 +00:00
nixpkgs-upkeep-bot ee4d3df08b python3Packages.jax: 0.2.27 -> 0.2.28 2022-02-02 10:05:04 -08:00
nixpkgs-upkeep-bot 53047281eb python3Packages.jax: 0.2.26 -> 0.2.27 2022-01-19 00:47:57 +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
Jonathan Ringer 685788d164
python3Packages.jax: 0.2.24 -> 0.2.25 2021-11-30 18:51:59 -08:00
Fabian Affolter ddf89d7d96 python3Packages.jax: 0.2.21 -> 0.2.24 2021-11-09 08:43:22 -08:00
Martin Weinelt 1b07699d46 python3Packages.jax: 0.2.19 -> 0.2.21 2021-10-11 01:22:03 +02:00
Samuel Ainsworth 6f44416cf2 python3Packages.jax: remove meta.description period 2021-09-01 21:04:02 +00:00
Samuel Ainsworth 426569a041 python3Packages.jax: init at 0.2.19 2021-08-22 20:39:04 +00:00