python3Packages.jax: 0.9.2 -> 0.10.0 (680d5961) · Commits · nix / nixpkgs

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

+2 −2

Original line number	Diff line number	Diff line
		@@ -50,8 +50,8 @@ buildPythonPackage (finalAttrs: {
		.${stdenv.hostPlatform.system};
		hash =
		{
		x86_64-linux = "sha256-U2owUpInbFdF77un61dXaEnFp8dzmKOp5h/TG69RAvA=";
		aarch64-linux = "sha256-VvSifl8ZypFMD0QCU5RpqpLQG/cTNqzQ7Y/dwgqRvI0=";
		x86_64-linux = "sha256-MJUVRT9Zyq2Vv3bIvGScJLwOPRLQe68895K+CCq97js=";
		aarch64-linux = "sha256-NqwxbFiKDgjrZPhoJwUmrco4VtS3CxN2C36bc76F0s8=";
		}
		.${stdenv.hostPlatform.system};
		};

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

+8 −8

Original line number	Diff line number	Diff line
		@@ -39,42 +39,42 @@ let
		"3.11-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_x86_64";
		dist = "cp311";
		hash = "sha256-1Vd82Ge9kmd2nkU7rYUNSAeoQ5a8l29jKlFe29d+SEs=";
		hash = "sha256-lXMXyv8eawBqE1UXM3cN8jWidHBkviawJ0YP46WPWO4=";
		};
		"3.11-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_aarch64";
		dist = "cp311";
		hash = "sha256-s5VfN10XkC8NJ+cFlnLNGWOlU0WVOkJpnk4HjOxyWtw=";
		hash = "sha256-GkpKVqgyrTDgHBQIWx+iRApV+kzggI5AXnNlDv+K+zE=";
		};
		"3.12-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_x86_64";
		dist = "cp312";
		hash = "sha256-iKVZCNd1sG3akqjE9MAVd44lulw2BbV/hLAAUvZujvE=";
		hash = "sha256-oWugDXI2YUWCfKOajISfn+4Rd9ZzjJ7f1ys6n+31zxg=";
		};
		"3.12-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_aarch64";
		dist = "cp312";
		hash = "sha256-sozPBbzAvHzLy9Mm2AKEZXTPbaA5FY52FHvZb1xvEYk=";
		hash = "sha256-6vGmGVqg60i0IMcfIrktGXeNlIAkl3yuRkEnnV9sH4E=";
		};
		"3.13-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_x86_64";
		dist = "cp313";
		hash = "sha256-vX3+0Xv6nQ4wFvjCpnZ8dHnZHhvf33kW6ysHQ1zEZY4=";
		hash = "sha256-7QdNN5Z7zmHBuv3fndCoMCMGwS+QcAylrfo9CP3Uruc=";
		};
		"3.13-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_aarch64";
		dist = "cp313";
		hash = "sha256-uaJwhdiTzFnCsoaxeJdV+Rzz6rHeobW+nmMvTJc5og4=";
		hash = "sha256-dHnmfZ2Y1KD1JflPi4GIDcwFhjH23hGLkd6R2xtAle0=";
		};
		"3.14-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_x86_64";
		dist = "cp314";
		hash = "sha256-U1F0LA/LIdqeCUoZZasg/eUlhih392kYpJCxtWZk1To=";
		hash = "sha256-96ahiBNJ3DnyRIGioBn/KUuUkJuAqIge/iHUW2pfaR0=";
		};
		"3.14-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_aarch64";
		dist = "cp314";
		hash = "sha256-MyEmmeG7sb7V0q4Urp/3Kh7tLQkqUearzAJ4prK4KHQ=";
		hash = "sha256-iIb8V5df7gG+FyX4mLaSkvYVtwgwrvXC+95jgzNha10=";
		};
		};
		in

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

+3 −11

Original line number	Diff line number	Diff line
		@@ -6,7 +6,6 @@
		lapack,
		buildPythonPackage,
		fetchFromGitHub,
		fetchpatch2,
		cudaSupport ? config.cudaSupport,

		# build-system
		@@ -41,25 +40,18 @@ let
		in
		buildPythonPackage (finalAttrs: {
		pname = "jax";
		version = "0.9.2";
		version = "0.10.0";
		pyproject = true;
		__structuredAttrs = true;

		src = fetchFromGitHub {
		owner = "google";
		repo = "jax";
		# google/jax contains tags for jax and jaxlib. Only use jax tags!
		tag = "jax-v${finalAttrs.version}";
		hash = "sha256-/vLCTAF46M1H0Q64RLM7+IFMofmjZmZ4IFzvm/y7zkg=";
		hash = "sha256-/RCihrjONN/+QwyQRNEmlIa7JsCLzz+SkBe5sd+ThgU=";
		};

		patches = [
		# https://github.com/jax-ml/jax/pull/32840
		(fetchpatch2 {
		url = "https://github.com/Prince213/jax/commit/af5c211d49f3b99447db2252d2cc2b8e0fb54d1c.patch?full_index=1";
		hash = "sha256-ijEd+MDe91qyYfE+aMzR5rNmTeGadin6Io8PIfJWc3o=";
		})
		];

		build-system = [ setuptools ];

		# The version is automatically set to ".dev" if this variable is not set.

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

+15 −13

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

		let
		version = "0.9.2";
		version = "0.10.0";
		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,65 +49,65 @@ let
		"3.11-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_x86_64";
		dist = "cp311";
		hash = "sha256-msmVtLoaru2uDWnzGZh9UV3K7NRQW2QrYxL54VQ5NR8=";
		hash = "sha256-m+IpmTpB5bK4TyNOzBml3gLzXt2xGVzwJ71TnhYB4V0=";
		};
		"3.11-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_aarch64";
		dist = "cp311";
		hash = "sha256-MG3lSh3nOGyAbHI+NWzjMtkj73SPinLWdP77dIEh1Nw=";
		hash = "sha256-PblOvIWTddlV3jUEGCrdfOFzPOPTDBXg7wMWAstRpVk=";
		};
		"3.11-aarch64-darwin" = getSrcFromPypi {
		platform = "macosx_11_0_arm64";
		dist = "cp311";
		hash = "sha256-eF8XfD63jLfceXxV7VxLYxIUGEXJpoaVfkhLrL/OXog=";
		hash = "sha256-J3Ay6fB0w/1f/R4MsD1P5m4nLeRyZnzbxBitmbIbZGo=";
		};

		"3.12-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_x86_64";
		dist = "cp312";
		hash = "sha256-iLJ2px9PIHGx/S6SKr/WfIfGl3pVGhA2/rzqeNXvfiI=";
		hash = "sha256-sL+4ZaB98ubXQYwLDCkt0pS1UAUjsd1YcrGA2yqkgNQ=";
		};
		"3.12-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_aarch64";
		dist = "cp312";
		hash = "sha256-/vAthGhjtybnJFKZOIOoWW6sMl8ioux+qSHaD7xVCbQ=";
		hash = "sha256-qh1w8aTifrQDZU5x4vso1XhtPpt3/BhH6MU4mICSfKQ=";
		};
		"3.12-aarch64-darwin" = getSrcFromPypi {
		platform = "macosx_11_0_arm64";
		dist = "cp312";
		hash = "sha256-l8L75Yy+5KJ9lMpzXXCdIxspmrbtizsQdfUthk39MsE=";
		hash = "sha256-fB2bRjMnx6IzPyEBFOywTyj+/FG6gjOoWiKAzOdb20I=";
		};

		"3.13-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_x86_64";
		dist = "cp313";
		hash = "sha256-ttUAPjrdXDRqNK6e3EcFjLwttgyO1cUAllIhdtrwHJ8=";
		hash = "sha256-0wPcMbZei3k9VgD4GxWDvgPcm4dqTBCz4lm2YJocvjs=";
		};
		"3.13-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_aarch64";
		dist = "cp313";
		hash = "sha256-vvYe7zbtOM7BBp6pc/iK+eAzNeiE9lAew/5/YiKhVVs=";
		hash = "sha256-bY14twcLNOTFu6X34Qkn5/Sqybab4X6bCliYVTpDOPM=";
		};
		"3.13-aarch64-darwin" = getSrcFromPypi {
		platform = "macosx_11_0_arm64";
		dist = "cp313";
		hash = "sha256-UqADJQj4z1eRx6e+4UJTHucGw8BVGBF/sLbujV4X/ec=";
		hash = "sha256-OEY1//VYmaKVu8gu5sb3c6MA54fcRyypK755q/qsg2k=";
		};

		"3.14-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_x86_64";
		dist = "cp314";
		hash = "sha256-PXFRFApJNvMhiy0bE0PdI3vShlz1FEKIS22C/ohKPec=";
		hash = "sha256-KkLPBMD4i8A7FQoX+n3bsvQOCWZn7IobhA7YeRPm5zU=";
		};
		"3.14-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux_2_27_aarch64";
		dist = "cp314";
		hash = "sha256-vkYnxC1Erdf+F9KE71ef+NFZ48tpR/ZDd1jzQXfoeOY=";
		hash = "sha256-rUfgckMJeewhY3qkh9TcRkAouOm+JyaPN95pU2x240E=";
		};
		"3.14-aarch64-darwin" = getSrcFromPypi {
		platform = "macosx_11_0_arm64";
		dist = "cp314";
		hash = "sha256-urFo0lVVRkRhvQdzI0hPaQxHHmnOiww5o5+4Gz46i/A=";
		hash = "sha256-mLJmcpQ2cnQoc/ZbwDIWgZ/FUyXJnxRlkNAHwBcr/zA=";
		};
		};
		in
		@@ -116,6 +116,8 @@ buildPythonPackage {
		inherit version;
		format = "wheel";

		__structuredAttrs = true;

		# See https://discourse.nixos.org/t/ofborg-does-not-respect-meta-platforms/27019/6.
		src = (
		srcs."${pythonVersion}-${stdenv.hostPlatform.system}"

Admin message