blob: ba3eda8cafbb43db2d521da21705932edbd57c9d (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
{
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;
};
}
|