0
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-07-09 12:05:50 +03:00

python311Packages.jax: 0.4.21 -> 0.4.23

Changelog: https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-23-dec-13-2023
This commit is contained in:
Gaetan Lepage 2023-12-14 10:18:12 +01:00
parent 601e70eed8
commit d0b8914d4a

View file

@ -6,6 +6,7 @@
, fetchFromGitHub , fetchFromGitHub
, jaxlib , jaxlib
, jaxlib-bin , jaxlib-bin
, hypothesis
, lapack , lapack
, matplotlib , matplotlib
, ml-dtypes , ml-dtypes
@ -27,7 +28,7 @@ let
in in
buildPythonPackage rec { buildPythonPackage rec {
pname = "jax"; pname = "jax";
version = "0.4.21"; version = "0.4.23";
pyproject = true; pyproject = true;
disabled = pythonOlder "3.9"; disabled = pythonOlder "3.9";
@ -37,7 +38,7 @@ buildPythonPackage rec {
repo = "jax"; repo = "jax";
# google/jax contains tags for jax and jaxlib. Only use jax tags! # google/jax contains tags for jax and jaxlib. Only use jax tags!
rev = "refs/tags/${pname}-v${version}"; rev = "refs/tags/${pname}-v${version}";
hash = "sha256-CMsW/t4/itJxN4pST8EKkN0ooHWdjRnLs073FwbXRJM="; hash = "sha256-PDa3yVH/sszGbWkVkJ+19FdOr3oqdYk+OdbeUTMTDuU=";
}; };
nativeBuildInputs = [ nativeBuildInputs = [
@ -59,6 +60,7 @@ buildPythonPackage rec {
] ++ lib.optional (pythonOlder "3.10") importlib-metadata; ] ++ lib.optional (pythonOlder "3.10") importlib-metadata;
nativeCheckInputs = [ nativeCheckInputs = [
hypothesis
jaxlib' jaxlib'
matplotlib matplotlib
pytestCheckHook pytestCheckHook