Commit 97f6797b authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python311Packages.jax: 0.4.19 -> 0.4.20 

Changelog: https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-20-nov-2-2023
parent 33de75c3

pkgs/development/python-modules/jax/default.nix

+6 −2
Original line number Diff line number Diff line
@@ -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) [