dev/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

python311Packages.flax: 0.8.3 -> 0.8.4

Browse Source 
Diff: https://github.com/google/flax/compare/refs/tags/v0.8.3...v0.8.4

Changelog: https://github.com/google/flax/releases/tag/v0.8.4
This commit is contained in:
Gaetan Lepage 2024-05-25 17:31:11 +02:00
parent 1a082d42b2
commit 5d96b117e1
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pkgs/development/python-modules/flax/default.nix
View File

@ -26,7 +26,7 @@
buildPythonPackage rec {
pname = "flax";
version = "0.8.3";
version = "0.8.4";
pyproject = true;
disabled = pythonOlder "3.9";
@ -35,7 +35,7 @@ buildPythonPackage rec {
owner = "google";
repo = "flax";
rev = "refs/tags/v${version}";
hash = "sha256-uDGTyksUZTTL6FiTJP+qteFLOjr75dcTj9yRJ6Jm8xU=";
hash = "sha256-ZwqKZdJ9LOfWTav5nE9xMsMw/DbryqQUuu5fqeugBzY=";
};
build-system = [
@ -86,7 +86,7 @@ buildPythonPackage rec {
# `tensorflow_datasets`, `vocabulary`) so the benefits of trying to run them
# would be limited anyway.
"examples/*"
"flax/experimental/nnx/examples/*"
"flax/nnx/examples/*"
# See https://github.com/google/flax/issues/3232.
"tests/jax_utils_test.py"
# Requires tree
@ -98,11 +98,11 @@ buildPythonPackage rec {
"test_overwrite_checkpoints0"
];
meta = with lib; {
meta = {
description = "Neural network library for JAX";
homepage = "https://github.com/google/flax";
changelog = "https://github.com/google/flax/releases/tag/v${version}";
license = licenses.asl20;
maintainers = with maintainers; [ ndl ];
license = lib.licenses.asl20;
maintainers = with lib.maintainers; [ ndl ];
};
}