pyrox.dev
lol
python311Packages.jax: 0.4.19 -> 0.4.20
Changelog: https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-20-nov-2-2023
+6
-2
+6
-2
pkgs/development/python-modules/jax/default.nix
+6
-2
pkgs/development/python-modules/jax/default.nix
···
27
27
in
28
28
buildPythonPackage rec {
29
29
pname = "jax";
30
-
version = "0.4.19";
30
+
version = "0.4.20";
31
31
pyproject = true;
32
32
33
33
disabled = pythonOlder "3.9";
···
37
37
repo = "jax";
38
38
# google/jax contains tags for jax and jaxlib. Only use jax tags!
39
39
rev = "refs/tags/${pname}-v${version}";
40
-
hash = "sha256-l5uLPqhg/hqtO9oJSaioow5cH/0jKHDVziGezkfnVcc=";
40
+
hash = "sha256-WLYXUtchOaA6SGnKuVhN9CmV06xMCLQTEuEtL13ttZU=";
41
41
};
42
42
43
43
nativeBuildInputs = [
···
108
108
"test_device_put"
109
109
"test_make_array_from_callback"
110
110
"test_make_array_from_single_device_arrays"
111
+
112
+
# Fails on some hardware due to some numerical error
113
+
# See https://github.com/google/jax/issues/18535
114
+
"testQdwhWithOnRankDeficientInput5"
111
115
];
112
116
113
117
disabledTestPaths = lib.optionals (stdenv.isDarwin && stdenv.isAarch64) [