pyrox.dev
python3Packages.jax: 0.3.16 -> 0.3.23
+4
-5
pkgs/development/python-modules/jax/default.nix
+4
-5
pkgs/development/python-modules/jax/default.nix
···
21
in
22
buildPythonPackage rec {
23
pname = "jax";
24
-
version = "0.3.16";
25
format = "setuptools";
26
27
disabled = pythonOlder "3.7";
···
30
owner = "google";
31
repo = pname;
32
rev = "jax-v${version}";
33
-
hash = "sha256-4idh7boqBXSO9vEHxEcrzXjBIrKmmXiCf6cXh7En1/I=";
34
};
35
36
# jaxlib is _not_ included in propagatedBuildInputs because there are
···
92
"tests/sparse_test.py"
93
];
94
95
-
pythonImportsCheck = [
96
-
"jax"
97
-
];
98
99
meta = with lib; {
100
description = "Differentiate, compile, and transform Numpy code";
···
21
in
22
buildPythonPackage rec {
23
pname = "jax";
24
+
version = "0.3.23";
25
format = "setuptools";
26
27
disabled = pythonOlder "3.7";
···
30
owner = "google";
31
repo = pname;
32
rev = "jax-v${version}";
33
+
hash = "sha256-ruXOwpBwpi1G8jgH9nhbWbs14JupwWkjh+Wzrj8HVU4=";
34
};
35
36
# jaxlib is _not_ included in propagatedBuildInputs because there are
···
92
"tests/sparse_test.py"
93
];
94
95
+
# As of 0.3.22, `import jax` does not work without jaxlib being installed.
96
+
pythonImportsCheck = [ ];
97
98
meta = with lib; {
99
description = "Differentiate, compile, and transform Numpy code";