Unverified Commit c340267c authored by Samuel Ainsworth's avatar Samuel Ainsworth Committed by GitHub
Browse files

Merge pull request #262149 from GaetanLepage/jax 

python310Packages.{jax,jaxlib,jaxlib-bin}: 0.4.18 -> 0.4.19
parents 46da009c 06eeedcf

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

+3 −3
Original line number Diff line number Diff line
@@ -15,16 +15,16 @@ 


buildPythonPackage rec {

  pname = "chex";

  version = "0.1.83";

  version = "0.1.84";

  format = "setuptools";



  disabled = pythonOlder "3.9";



  src = fetchFromGitHub {

    owner = "deepmind";

    repo = pname;

    repo = "chex";

    rev = "refs/tags/v${version}";

    hash = "sha256-iEachJf5NjOnkMWdP0aVQHWNPgUUBkMnzHKq3GP7t4w=";

    hash = "sha256-LsUMvSMVGjqZuFDcb+/61RtFxweeG6bSFzmJUUMv6rA=";

  };



  propagatedBuildInputs = [

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

+3 −3
Original line number Diff line number Diff line
@@ -27,17 +27,17 @@ let 
in

buildPythonPackage rec {

  pname = "jax";

  version = "0.4.18";

  version = "0.4.19";

  pyproject = true;



  disabled = pythonOlder "3.9";



  src = fetchFromGitHub {

    owner = "google";

    repo = pname;

    repo = "jax";

    # google/jax contains tags for jax and jaxlib. Only use jax tags!

    rev = "refs/tags/${pname}-v${version}";

    hash = "sha256-rDvWHa8jYCAA9iKbWaFUXdE/9L7AepFiNzmqOcc/090=";

    hash = "sha256-l5uLPqhg/hqtO9oJSaioow5cH/0jKHDVziGezkfnVcc=";

  };



  nativeBuildInputs = [

pkgs/development/python-modules/jaxlib/bin.nix

+5 −5
Original line number Diff line number Diff line
@@ -39,7 +39,7 @@ in 
assert cudaSupport -> lib.versionAtLeast cudatoolkit.version "11.1" && lib.versionAtLeast cudnn.version "8.2" && stdenv.isLinux;



let

  version = "0.4.18";

  version = "0.4.19";



  inherit (python) pythonVersion;
@@ -60,15 +60,15 @@ let 
    {

      "x86_64-linux" = getSrcFromPypi {

        platform = "manylinux2014_x86_64";

        hash = "sha256-MpNomovvSVx4N6gsowOLksTyEgTK261vSXMGxYqlVOE=";

        hash = "sha256-ksnY+CPEstact5lKjbSg+ZSPJtSt0Y0NFWEFufBCByk=";

      };

      "aarch64-darwin" = getSrcFromPypi {

        platform = "macosx_11_0_arm64";

        hash = "sha256-if/5O5DQVHFdsLw9O1creZBx5j8ftE7fsWMMX1NjHP0=";

        hash = "sha256-O7dHvdKLKfNELGfF4TKy7N5EX6Ca7Zu8OtLXWvFykR8=";

      };

      "x86_64-darwin" = getSrcFromPypi {

        platform = "macosx_10_14_x86_64";

        hash = "sha256-4NeHA/0SGdmHXyDGxpK7oJc7dE1meR4LPjzbIwxloqU=";

        hash = "sha256-gqKMUZSXrt8sQtTAoQbzAfCzO8gM9Y1/tZpuJVWyN0Y=";

      };

    };
@@ -78,7 +78,7 @@ let 
  # https://github.com/google/jax/issues/12879 as to why this specific URL is the correct index.

  gpuSrc = fetchurl {

    url = "https://storage.googleapis.com/jax-releases/cuda12/jaxlib-${version}+cuda12.cudnn89-cp310-cp310-manylinux2014_x86_64.whl";

    hash = "sha256-p6BNvhhRzVDQdpEoIRau5JovC+eDjlW3bXrahtsGvmI=";

    hash = "sha256-zfN0n31+5GohwBkeQrqHus4qOyhM/GEdqG6KUupCZ4o=";

  };



in

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

+5 −5
Original line number Diff line number Diff line
@@ -54,7 +54,7 @@ let 
  inherit (cudaPackages) backendStdenv cudatoolkit cudaFlags cudnn nccl;



  pname = "jaxlib";

  version = "0.4.18";

  version = "0.4.19";



  meta = with lib; {

    description = "JAX is Autograd and XLA, brought together for high-performance machine learning research.";
@@ -151,7 +151,7 @@ let 
      repo = "jax";

      # google/jax contains tags for jax and jaxlib. Only use jaxlib tags!

      rev = "refs/tags/${pname}-v${version}";

      hash = "sha256-rDvWHa8jYCAA9iKbWaFUXdE/9L7AepFiNzmqOcc/090=";

      hash = "sha256-l5uLPqhg/hqtO9oJSaioow5cH/0jKHDVziGezkfnVcc=";

    };



    nativeBuildInputs = [
@@ -264,10 +264,10 @@ let 
      ];



      sha256 = (if cudaSupport then {

        x86_64-linux = "sha256-0CfGWlwKsUFP1DHUN6+6wX3cHr5x3TE6NbqYlV5me1E=";

        x86_64-linux = "sha256-Z5cSgdRxdKxidaz4b1RlUF4rVcQiUTmQ1OorlBWlpt0=";

      } else {

        x86_64-linux = "sha256-sljmyIligXC7d9fdlpqR32xyMR0UslWs04gXJBD8FTA=";

        aarch64-linux = "sha256-eJ4KIkHdcA2EVvyBoNum2cOPcHPFoBOtUTAGufO8FJA=";

        x86_64-linux = "sha256-sn7p8FFHWIVdBWnsLsVj5jLiSaTlRm7s/qj2RqvQ3jU=";

        aarch64-linux = "sha256-oAYF5AeuPHTlwtpDMs2+tAhRAJH0yeSVnB7Ni7wmzS8=";

      }).${stdenv.system} or (throw "jaxlib: unsupported system: ${stdenv.system}");

    };

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

+2 −2
Original line number Diff line number Diff line
@@ -26,7 +26,7 @@ 


buildPythonPackage rec {

  pname = "pytensor";

  version = "2.17.2";

  version = "2.17.3";

  pyproject = true;



  disabled = pythonOlder "3.9";
@@ -35,7 +35,7 @@ buildPythonPackage rec { 
    owner = "pymc-devs";

    repo = "pytensor";

    rev = "refs/tags/rel-${version}";

    hash = "sha256-u1CbOjU3rQ6G3SSwYR3UlebymkupGMJWID4RH4v9PIk=";

    hash = "sha256-FufPCFzSjG8BrHes7t3XsdovX9gqUBG0gMDGKvkRkSA=";

  };



  postPatch = ''