diff options
author | Gaetan Lepage <gaetan@glepage.com> | 2023-11-04 10:16:39 +0100 |
---|---|---|
committer | Gaetan Lepage <gaetan@glepage.com> | 2023-11-15 07:55:10 +0100 |
commit | 97f6797b8316d8a16b09b77f341b233c22c205b1 (patch) | |
tree | 4f741afe2a698da4044472e76b64dc26442d0d6c | |
parent | 33de75c3670db00b1314e3e069f8fbdb3019eef9 (diff) |
python311Packages.jax: 0.4.19 -> 0.4.20
Changelog: https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-20-nov-2-2023
-rw-r--r-- | pkgs/development/python-modules/jax/default.nix | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/jax/default.nix b/pkgs/development/python-modules/jax/default.nix index d9293e0734801..203aa49db8fb6 100644 --- a/pkgs/development/python-modules/jax/default.nix +++ b/pkgs/development/python-modules/jax/default.nix @@ -27,7 +27,7 @@ let in buildPythonPackage rec { pname = "jax"; - version = "0.4.19"; + version = "0.4.20"; pyproject = true; disabled = pythonOlder "3.9"; @@ -37,7 +37,7 @@ buildPythonPackage rec { repo = "jax"; # google/jax contains tags for jax and jaxlib. Only use jax tags! rev = "refs/tags/${pname}-v${version}"; - hash = "sha256-l5uLPqhg/hqtO9oJSaioow5cH/0jKHDVziGezkfnVcc="; + hash = "sha256-WLYXUtchOaA6SGnKuVhN9CmV06xMCLQTEuEtL13ttZU="; }; nativeBuildInputs = [ @@ -108,6 +108,10 @@ buildPythonPackage rec { "test_device_put" "test_make_array_from_callback" "test_make_array_from_single_device_arrays" + + # Fails on some hardware due to some numerical error + # See https://github.com/google/jax/issues/18535 + "testQdwhWithOnRankDeficientInput5" ]; disabledTestPaths = lib.optionals (stdenv.isDarwin && stdenv.isAarch64) [ |