python310Packages.torchdiffeq: init at 0.2.3
This commit is contained in:
parent
2901b67d76
commit
3ba5ff8416
43
pkgs/development/python-modules/torchdiffeq/default.nix
Normal file
43
pkgs/development/python-modules/torchdiffeq/default.nix
Normal file
|
@ -0,0 +1,43 @@
|
||||||
|
{ lib
|
||||||
|
, buildPythonPackage
|
||||||
|
, fetchPypi
|
||||||
|
|
||||||
|
# dependencies
|
||||||
|
, torch
|
||||||
|
, scipy
|
||||||
|
|
||||||
|
# tests
|
||||||
|
, pytestCheckHook
|
||||||
|
}:
|
||||||
|
|
||||||
|
buildPythonPackage rec {
|
||||||
|
pname = "torchdiffeq";
|
||||||
|
version = "0.2.3";
|
||||||
|
format = "setuptools";
|
||||||
|
|
||||||
|
src = fetchPypi {
|
||||||
|
inherit pname version;
|
||||||
|
hash = "sha256-/nX0NLkJCsDCdwLgK+0hRysPhwNb5lgfUe3F1AE+oxo=";
|
||||||
|
};
|
||||||
|
|
||||||
|
propagatedBuildInputs = [
|
||||||
|
torch
|
||||||
|
scipy
|
||||||
|
];
|
||||||
|
|
||||||
|
pythonImportsCheck = [ "torchdiffeq" ];
|
||||||
|
|
||||||
|
# no tests in sdist, no tags on git
|
||||||
|
doCheck = false;
|
||||||
|
|
||||||
|
nativeCheckInputs = [
|
||||||
|
pytestCheckHook
|
||||||
|
];
|
||||||
|
|
||||||
|
meta = with lib; {
|
||||||
|
description = "Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation";
|
||||||
|
homepage = "https://github.com/rtqichen/torchdiffeq";
|
||||||
|
license = licenses.mit;
|
||||||
|
maintainers = teams.tts.members;
|
||||||
|
};
|
||||||
|
}
|
|
@ -12149,6 +12149,8 @@ self: super: with self; {
|
||||||
|
|
||||||
torchaudio-bin = callPackage ../development/python-modules/torchaudio/bin.nix { };
|
torchaudio-bin = callPackage ../development/python-modules/torchaudio/bin.nix { };
|
||||||
|
|
||||||
|
torchdiffeq = callPackage ../development/python-modules/torchdiffeq { };
|
||||||
|
|
||||||
torchgpipe = callPackage ../development/python-modules/torchgpipe { };
|
torchgpipe = callPackage ../development/python-modules/torchgpipe { };
|
||||||
|
|
||||||
torchmetrics = callPackage ../development/python-modules/torchmetrics { };
|
torchmetrics = callPackage ../development/python-modules/torchmetrics { };
|
||||||
|
|
Loading…
Reference in a new issue