tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
python3Packages.jax[lib]: 0.6.1 -> 0.6.2
Diff: https://github.com/jax-ml/jax/compare/jax-v0.6.1...jax-v0.6.2
Changelog: https://github.com/jax-ml/jax/releases/tag/jax-v0.6.2
+25
-25
+2
-2
pkgs/development/python-modules/jax-cuda12-pjrt/default.nix
+2
-2
pkgs/development/python-modules/jax-cuda12-pjrt/default.nix
···
51
51
.${stdenv.hostPlatform.system};
52
52
hash =
53
53
{
54
-
x86_64-linux = "sha256-TJfRClqawJ+gAVaMrDtxUBTo27ws2GdjdT9Y5acwwzM=";
55
-
aarch64-linux = "sha256-lnB2z7by4zlZ5zdmY1maoMEcwO3o8vUaIG2godQixrs=";
54
+
x86_64-linux = "sha256-jNnq15SOosd4pQj+9dEVnot6v0/MxwN8P+Hb/NlQEtw=";
55
+
aarch64-linux = "sha256-IvrwINLo98oeKRVjMkH333Z4tzxwePXwsvETJIM3994=";
56
56
}
57
57
.${stdenv.hostPlatform.system};
58
58
};
+8
-8
pkgs/development/python-modules/jax-cuda12-plugin/default.nix
+8
-8
pkgs/development/python-modules/jax-cuda12-plugin/default.nix
···
40
40
"3.10-x86_64-linux" = getSrcFromPypi {
41
41
platform = "manylinux2014_x86_64";
42
42
dist = "cp310";
43
-
hash = "sha256-t3gE4OTZI605kJCV/3wbcj6sbz7l+f/LgFl7qGe1crg=";
43
+
hash = "sha256-F1H4iYkmmzzbDf5PewcqZEIUmBjJvJjDo5XIrK+RCnk=";
44
44
};
45
45
"3.10-aarch64-linux" = getSrcFromPypi {
46
46
platform = "manylinux2014_aarch64";
47
47
dist = "cp310";
48
-
hash = "sha256-2p99ySQ+wo4DwOOjmFK0JG+pz8Pc1R5Chtggl/XGlcA=";
48
+
hash = "sha256-vFw6ddBVGbTTJuRmnQ960P4PCs+HX5MT2RN0jMylqeo=";
49
49
};
50
50
"3.11-x86_64-linux" = getSrcFromPypi {
51
51
platform = "manylinux2014_x86_64";
52
52
dist = "cp311";
53
-
hash = "sha256-H7+NS0JFVEOgia/RqI+xBqUboQdfxohLM53JZXHFthc=";
53
+
hash = "sha256-CJbLswjZUpHiBc2J0lQCne46HfQ9ZumDEzGpr9LSeHA=";
54
54
};
55
55
"3.11-aarch64-linux" = getSrcFromPypi {
56
56
platform = "manylinux2014_aarch64";
57
57
dist = "cp311";
58
-
hash = "sha256-QlzPE8vdRnixEJ+EOYgVelnk9Nm8KYIFrLFt8EijHDg=";
58
+
hash = "sha256-LNjieaWaOLoMl4qDHhOt627p5Fcvujh8eXW6OtU13Tg=";
59
59
};
60
60
"3.12-x86_64-linux" = getSrcFromPypi {
61
61
platform = "manylinux2014_x86_64";
62
62
dist = "cp312";
63
-
hash = "sha256-GIXxW+OPrszPvySxhP/cHQ02Nxfq3SU01XWcDT0K9SM=";
63
+
hash = "sha256-/r0Jn5cNNQ64+losmi+0sOp7PWqJ3xSWZj7fp6/lkOU=";
64
64
};
65
65
"3.12-aarch64-linux" = getSrcFromPypi {
66
66
platform = "manylinux2014_aarch64";
67
67
dist = "cp312";
68
-
hash = "sha256-u2SgyAH5OnGKZU38aXQvL9YKJgdDEiBOvfT+QD2eK8Q=";
68
+
hash = "sha256-bJsALROx/LlANxPu3Th2oietH/vfs4EbH5+Jr0wlpfc=";
69
69
};
70
70
"3.13-x86_64-linux" = getSrcFromPypi {
71
71
platform = "manylinux2014_x86_64";
72
72
dist = "cp313";
73
-
hash = "sha256-KjV43At9RMwbAjOw/nrXZCZTgQldfqxkxWvQGzS+dvI=";
73
+
hash = "sha256-20xhA8kS2M0a35TDTTE7tHYMp/AciXynzWLmXyeZQZk=";
74
74
};
75
75
"3.13-aarch64-linux" = getSrcFromPypi {
76
76
platform = "manylinux2014_aarch64";
77
77
dist = "cp313";
78
-
hash = "sha256-uL/3pfx6QWcX4dWdqXKKH3qtB6i2WvoPhpYtQ+0OZU8=";
78
+
hash = "sha256-dz76i1WoN0BsVh8O8CFE3akBkYEZN2DsVBnuyd0rmqw=";
79
79
};
80
80
};
81
81
in
+2
-2
pkgs/development/python-modules/jax/default.nix
+2
-2
pkgs/development/python-modules/jax/default.nix
···
40
40
in
41
41
buildPythonPackage rec {
42
42
pname = "jax";
43
-
version = "0.6.1";
43
+
version = "0.6.2";
44
44
pyproject = true;
45
45
46
46
src = fetchFromGitHub {
···
48
48
repo = "jax";
49
49
# google/jax contains tags for jax and jaxlib. Only use jax tags!
50
50
tag = "jax-v${version}";
51
-
hash = "sha256-Am+ksPC4U3vL5LGmePzSaMSwWJOCcVrC+DFkJzJP+1U=";
51
+
hash = "sha256-MTgpwpJWxULCiZhDG+MFpOp8ZHoj1ZDmOD05OaGfXhM=";
52
52
};
53
53
54
54
build-system = [ setuptools ];
+13
-13
pkgs/development/python-modules/jaxlib/bin.nix
+13
-13
pkgs/development/python-modules/jaxlib/bin.nix
···
18
18
}:
19
19
20
20
let
21
-
version = "0.6.1";
21
+
version = "0.6.2";
22
22
inherit (python) pythonVersion;
23
23
24
24
# As of 2023-06-06, google/jax upstream is no longer publishing CPU-only wheels to their GCS bucket. Instead the
···
49
49
"3.10-x86_64-linux" = getSrcFromPypi {
50
50
platform = "manylinux2014_x86_64";
51
51
dist = "cp310";
52
-
hash = "sha256-euWBWtpxtpUyzkQ6ERYKPtJcZ+gqKUoNia+dTSdClDQ=";
52
+
hash = "sha256-wIeg62+39vj1TVb0cwMo395QQN07Xd+oEOfCjqcQK0I=";
53
53
};
54
54
"3.10-aarch64-linux" = getSrcFromPypi {
55
55
platform = "manylinux2014_aarch64";
56
56
dist = "cp310";
57
-
hash = "sha256-gQbcMW60QNB7nUYooMjirPdtpWBnQsn1wzEEqqd7CsI=";
57
+
hash = "sha256-QgXQmM6O+19/4v5QmLrmA2CU3I2IKfXg4NepsVUyYzY=";
58
58
};
59
59
"3.10-aarch64-darwin" = getSrcFromPypi {
60
60
platform = "macosx_11_0_arm64";
61
61
dist = "cp310";
62
-
hash = "sha256-J3zH6dZX0Ik6VZJhJ3s+rpFq1/pz4wCmKSYftTfcoPE=";
62
+
hash = "sha256-2kYBsrXcjCPWr7KT6s+5rsTh0YccsvKcWhUdED5zsPg=";
63
63
};
64
64
65
65
"3.11-x86_64-linux" = getSrcFromPypi {
66
66
platform = "manylinux2014_x86_64";
67
67
dist = "cp311";
68
-
hash = "sha256-EfzEscdBoeAFfy/6d9WoK/5+6Xw4ZO2I32dJPnibkXM=";
68
+
hash = "sha256-M11+NRXOeLUqQQE29Gqkp+oU0OfWQPNOHhN0CVVK0Kw=";
69
69
};
70
70
"3.11-aarch64-linux" = getSrcFromPypi {
71
71
platform = "manylinux2014_aarch64";
72
72
dist = "cp311";
73
-
hash = "sha256-WpDufFmywAdzAm+/kYJpx6hnamqBo0oDr5GffXvc6ag=";
73
+
hash = "sha256-Eern4FvFp5h12jYySvue3dS66u8qA4bK9tTzcgua7yg=";
74
74
};
75
75
"3.11-aarch64-darwin" = getSrcFromPypi {
76
76
platform = "macosx_11_0_arm64";
77
77
dist = "cp311";
78
-
hash = "sha256-ArrFFTOJ8BYWUWqf0dzWA40j7lBoHawU5N28Q8yzEzo=";
78
+
hash = "sha256-ogj/YcWBKNMGu05a0IWL0rCWDywcEK1CxUj3SmDAAg4=";
79
79
};
80
80
81
81
"3.12-x86_64-linux" = getSrcFromPypi {
82
82
platform = "manylinux2014_x86_64";
83
83
dist = "cp312";
84
-
hash = "sha256-0DkSRGhWW785NjsVBMGQ5nGeaviaeUje4lbx3ugTu5Q=";
84
+
hash = "sha256-8d0JtIGpPB1MdQAT9Gf3QZRJO6e9KfzU0c7BbjohT2U=";
85
85
};
86
86
"3.12-aarch64-linux" = getSrcFromPypi {
87
87
platform = "manylinux2014_aarch64";
88
88
dist = "cp312";
89
-
hash = "sha256-tYwp/nR2IrcJRuqHgjrTkgLMg9o9k6UpO0Mhc7c4qGg=";
89
+
hash = "sha256-h+wtycPtmrk27shTUWDF+9LISZSFWfHF2qdfY/q+WUI=";
90
90
};
91
91
"3.12-aarch64-darwin" = getSrcFromPypi {
92
92
platform = "macosx_11_0_arm64";
93
93
dist = "cp312";
94
-
hash = "sha256-4UGVwj7s1VmmHDECe0Fy6RLlpQ9jAyCRj/366DCQylo=";
94
+
hash = "sha256-NNimhKi+lJ3YfdSsyXEBtBBqDcmtFR7IkdoHIxmle5k=";
95
95
};
96
96
97
97
"3.13-x86_64-linux" = getSrcFromPypi {
98
98
platform = "manylinux2014_x86_64";
99
99
dist = "cp313";
100
-
hash = "sha256-5zS+cP4+H6KjFBU2JyEYnZdNEKZrD1OWyEWFWH0QGxU=";
100
+
hash = "sha256-+UFj8UyP07qTrhS2Maus8UywMbugtZE4hpmEtNEDdfg=";
101
101
};
102
102
"3.13-aarch64-linux" = getSrcFromPypi {
103
103
platform = "manylinux2014_aarch64";
104
104
dist = "cp313";
105
-
hash = "sha256-0MNDxRsQUlk+22A931jPf5iBKylRrmxFvW6T4+Hy9iE=";
105
+
hash = "sha256-cEmIN8r1OL1Fj/aFjIv9QE24IBWrqPZjZwGX+pkA/wI=";
106
106
};
107
107
"3.13-aarch64-darwin" = getSrcFromPypi {
108
108
platform = "macosx_11_0_arm64";
109
109
dist = "cp313";
110
-
hash = "sha256-9Mp12dR6LpAJmt/t4OnJJrg+9wPTSbMom4yI6GHAnl0=";
110
+
hash = "sha256-v/Z7GIEzzh8BEcexY6wyH9ZGtZ7SIepIkGPi4PhcuWc=";
111
111
};
112
112
};
113
113
in