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:
parent
601e70eed8
commit
d0b8914d4a
1 changed files with 4 additions and 2 deletions
|
@ -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
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue