nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-02-16 17:14:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #322609 from GaetanLepage/flax

Browse Source 
python311Packages.flax: 0.8.4 -> 0.8.5
This commit is contained in:
Gaétan Lepage 2024-06-26 21:53:15 +02:00 committed by GitHub
commit 4584edc9ac
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 21 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,32 +1,40 @@
{
lib,
buildPythonPackage,
cloudpickle,
einops,
pythonOlder,
fetchFromGitHub,
jax,
# build-system
jaxlib,
keras,
matplotlib,
pythonRelaxDepsHook,
setuptools-scm,
# dependencies
jax,
msgpack,
numpy,
optax,
orbax-checkpoint,
pytest-xdist,
pytestCheckHook,
pythonOlder,
pythonRelaxDepsHook,
pyyaml,
rich,
setuptools-scm,
tensorflow,
tensorstore,
typing-extensions,
# checks
cloudpickle,
einops,
keras,
pytest-xdist,
pytestCheckHook,
tensorflow,
# optional-dependencies
matplotlib,
}:
buildPythonPackage rec {
pname = "flax";
version = "0.8.4";
version = "0.8.5";
pyproject = true;
disabled = pythonOlder "3.9";
@ -35,7 +43,7 @@ buildPythonPackage rec {
owner = "google";
repo = "flax";
rev = "refs/tags/v${version}";
hash = "sha256-ZwqKZdJ9LOfWTav5nE9xMsMw/DbryqQUuu5fqeugBzY=";
hash = "sha256-6WOFq0758gtNdrlWqSQBlKmWVIGe5e4PAaGrvHoGjr0=";
};
build-system = [