python3Packages.jax: 0.8.0 -> 0.8.1 (#463012) (906352d5) · Commits · nix / nixpkgs

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

+5 −0

Original line number	Diff line number	Diff line
		@@ -57,6 +57,11 @@ buildPythonPackage rec {
		pytestCheckHook
		];

		pytestFlags = [
		# DeprecationWarning: The default axis_types will change in JAX v0.9.0 to jax.sharding.AxisType.Explicit.
		"-Wignore::DeprecationWarning"
		];

		disabledTests = [
		# Failed: DID NOT WARN. No warnings of type (<class 'Warning'>,) were emitted.
		"test_jax_transform_warn"

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

+2 −2

Original line number	Diff line number	Diff line
		@@ -50,8 +50,8 @@ buildPythonPackage rec {
		.${stdenv.hostPlatform.system};
		hash =
		{
		x86_64-linux = "sha256-npnD5Ub2vgD/xqTt0s9bdtTFmqNKjYOW2COauP2Ov9M=";
		aarch64-linux = "sha256-UAu8W9mrvcyyOP4TgbkAzVdyZFANFHA1H2jl29tSNcI=";
		x86_64-linux = "sha256-RStw7hDLmsXX38pV/7zbibbIvGunCkWvfEkNHc6pjrc=";
		aarch64-linux = "sha256-pjHQaJkDNUr9ez0uxZW32gamIwp22gD/lUj1QrIbYlA=";
		}
		.${stdenv.hostPlatform.system};
		};

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

+6 −6

Original line number	Diff line number	Diff line
		@@ -39,32 +39,32 @@ let
		"3.11-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_x86_64";
		dist = "cp311";
		hash = "sha256-cL6wFZM7RyEDSBonUh1NtBz1kgCHBfRBLL86HsrzXC4=";
		hash = "sha256-R5ykONVVAk2sjdEFg3Hvv49HmBmnCuohPz8wN+zpnXQ=";
		};
		"3.11-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_aarch64";
		dist = "cp311";
		hash = "sha256-x+pdt1IZS4o/v0xWS7e7wk8k4HEJu1nmWf1LETFLunY=";
		hash = "sha256-szg73AufYmDZrcTKDR9ovyQRWN/mnXJrJnsGgTgup6c=";
		};
		"3.12-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_x86_64";
		dist = "cp312";
		hash = "sha256-VVbj9Iy+HfYI2dDTQAMIZJ3eyqjQ4kOUkMb4i0GAMcs=";
		hash = "sha256-tgvwu9okzsb6cRcL1pthM1nwGjdtjgn+NL9n7MmjFk8=";
		};
		"3.12-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_aarch64";
		dist = "cp312";
		hash = "sha256-SdBMFULqpAi3utAXASC82oMqAvuUT8iT3OiCbublpJM=";
		hash = "sha256-OFAB9W+FKVnwYa4VrRV8OcxEccjR0lRN/D+AVoSsIhM=";
		};
		"3.13-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_x86_64";
		dist = "cp313";
		hash = "sha256-Rw6W5+pNsPN1jabn0VxF6BAmYpJP46hHIV7mXKySxo0=";
		hash = "sha256-c0LIgQzJR9548oxyh6MLLiAbD1FXhUPdJVNpK3mkmUI=";
		};
		"3.13-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_aarch64";
		dist = "cp313";
		hash = "sha256-IGmVNdeaLM+X8SEGC2BAUzhvQRzLMMElxmtqWgJsiSE=";
		hash = "sha256-2nwPLvHGl/mt5Rpxz60hHiv/JUB6aFXd3eNywBkPxGg=";
		};
		};
		in

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

+2 −2

Original line number	Diff line number	Diff line
		@@ -41,7 +41,7 @@ let
		in
		buildPythonPackage rec {
		pname = "jax";
		version = "0.8.0";
		version = "0.8.1";
		pyproject = true;

		src = fetchFromGitHub {
		@@ -49,7 +49,7 @@ buildPythonPackage rec {
		repo = "jax";
		# google/jax contains tags for jax and jaxlib. Only use jax tags!
		tag = "jax-v${version}";
		hash = "sha256-Ilcp4WW65SyqrqDGBRdnB25m7OCbrsfdtxWvl0uTjNw=";
		hash = "sha256-e1XqcecIl8TGOZmA1FuWg8uwVUvgSdp8uYltXm9cJpQ=";
		};

		patches = [

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

+10 −10

Original line number	Diff line number	Diff line
		@@ -18,7 +18,7 @@
		}:

		let
		version = "0.8.0";
		version = "0.8.1";
		inherit (python) pythonVersion;

		# As of 2023-06-06, google/jax upstream is no longer publishing CPU-only wheels to their GCS bucket. Instead the
		@@ -49,49 +49,49 @@ let
		"3.11-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_x86_64";
		dist = "cp311";
		hash = "sha256-/1Pouvl49rfEB2IVr3jwupacrENO0vclZdh+OMI/AOc=";
		hash = "sha256-IvSJ+1yL4Np75eSVehCTazdgoWlmj4slxdCcUcPvR/Y=";
		};
		"3.11-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_aarch64";
		dist = "cp311";
		hash = "sha256-Qa693vZ6VVpt4XQnpOZs5gpSioFYR+Ldltq85Xn3rPg=";
		hash = "sha256-/zK2Mg1ykTHvryKTmCW1LXWVfITDKvKwsb2zPPJ7p18=";
		};
		"3.11-aarch64-darwin" = getSrcFromPypi {
		platform = "macosx_11_0_arm64";
		dist = "cp311";
		hash = "sha256-u2AqjCTGFMuMpu7tPnCnM9k5nGoviJAKAlJiPNZydrU=";
		hash = "sha256-hlrdVhOYg0BfPxXJsN5qZKuPSqVJ3/GWty28hr5szB8=";
		};

		"3.12-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_x86_64";
		dist = "cp312";
		hash = "sha256-LIZ1v4bjka/k+NhjCAvhoCTXNN/T3RN/eqjn8iCRrc0=";
		hash = "sha256-r0kkGJ/FO2kjdxW1bry/xxu5HKFhhBQ9zvDUMMgXPeY=";
		};
		"3.12-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_aarch64";
		dist = "cp312";
		hash = "sha256-2D/4zxsHApljnNpPhCdwf2kFHchCHln7tzMFUjk3Vw0=";
		hash = "sha256-vtHpSujHwWvKRHbY1/WC8NGhAqTmnDqb0gaaDcQidKk=";
		};
		"3.12-aarch64-darwin" = getSrcFromPypi {
		platform = "macosx_11_0_arm64";
		dist = "cp312";
		hash = "sha256-9gqsD2Tp5wpc7zQf4pJoRRhpVRTHGtAANndLvtX3MS4=";
		hash = "sha256-iL3g9TXu6maJ4M1X1At2YNUgaslcfULglWKhCbljpJ8=";
		};

		"3.13-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_x86_64";
		dist = "cp313";
		hash = "sha256-Zsb1dvVKY+0FL1xGm+9NtyP18FC4OewMQpVzARNBvVg=";
		hash = "sha256-0kW9aieccspfeW34TN1k18nIq8S42JrfSs9FiY2rlYs=";
		};
		"3.13-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_aarch64";
		dist = "cp313";
		hash = "sha256-s+rFA7kP/sxo8R+hIhM+7yxixTbbKOgB5DbX56m2e/g=";
		hash = "sha256-vWl8FxrOHi6dbtkQp484WzxAlc7ikLAlWqWISPKs3qs=";
		};
		"3.13-aarch64-darwin" = getSrcFromPypi {
		platform = "macosx_11_0_arm64";
		dist = "cp313";
		hash = "sha256-X88zpWOfjxZKRzqceKH6Cy4VrD/L7NbZaqD4i/Jeprs=";
		hash = "sha256-oDSfboF53Il9M665DsZrSoBBMw+7uo0HHcYWfNInFTk=";
		};
		};
		in