nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-11-26 00:43:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

python312Packages.oryx: init at 0.2.7

Browse Source
This commit is contained in:
Gaetan Lepage 2024-09-30 07:52:09 +02:00 committed by Gaétan Lepage
parent 441ac0c553
commit 73187eaa33
2 changed files with 55 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
pkgs/development/python-modules/oryx/default.nix Normal file
View File

@ -0,0 +1,53 @@
{
lib,
buildPythonPackage,
fetchFromGitHub,
# build-system
poetry-core,
# dependencies
jax,
jaxlib,
tensorflow-probability,
# tests
inference-gym,
pytestCheckHook,
}:
buildPythonPackage rec {
pname = "oryx";
version = "0.2.7";
pyproject = true;
src = fetchFromGitHub {
owner = "jax-ml";
repo = "oryx";
rev = "refs/tags/v${version}";
hash = "sha256-1n7ogGuFNAeOyXWe0/pAouhg2+aA3MXxlCcsrfqRTdU=";
};
build-system = [ poetry-core ];
dependencies = [
jax
jaxlib
tensorflow-probability
];
pythonImportsCheck = [ "oryx" ];
nativeCheckInputs = [
inference-gym
pytestCheckHook
];
meta = {
description = "Library for probabilistic programming and deep learning built on top of Jax";
homepage = "https://github.com/jax-ml/oryx";
changelog = "https://github.com/jax-ml/oryx/releases/tag/v${version}";
license = lib.licenses.asl20;
maintainers = with lib.maintainers; [ GaetanLepage ];
};
}

2
pkgs/top-level/python-packages.nix
View File

@ -9462,6 +9462,8 @@ self: super: with self; {
orvibo = callPackage ../development/python-modules/orvibo { };
oryx = callPackage ../development/python-modules/oryx { };
os-client-config = callPackage ../development/python-modules/os-client-config { };
os-service-types = callPackage ../development/python-modules/os-service-types { };