tjh.dev
+63
pkgs/development/python-modules/optimistix/default.nix
+63
pkgs/development/python-modules/optimistix/default.nix
···
1
+
{
2
+
lib,
3
+
buildPythonPackage,
4
+
fetchFromGitHub,
5
+
6
+
# build-system
7
+
hatchling,
8
+
9
+
# dependencies
10
+
equinox,
11
+
jax,
12
+
jaxtyping,
13
+
lineax,
14
+
typing-extensions,
15
+
16
+
# checks
17
+
beartype,
18
+
jaxlib,
19
+
optax,
20
+
pytestCheckHook,
21
+
pytest-xdist,
22
+
}:
23
+
24
+
buildPythonPackage rec {
25
+
pname = "optimistix";
26
+
version = "0.0.8";
27
+
pyproject = true;
28
+
29
+
src = fetchFromGitHub {
30
+
owner = "patrick-kidger";
31
+
repo = "optimistix";
32
+
rev = "refs/tags/v${version}";
33
+
hash = "sha256-0ehNApiBoiAb8LFBW81ZCRPsjTVQG8zRTVSAp7oHF8w=";
34
+
};
35
+
36
+
build-system = [ hatchling ];
37
+
38
+
dependencies = [
39
+
equinox
40
+
jax
41
+
jaxtyping
42
+
lineax
43
+
typing-extensions
44
+
];
45
+
46
+
pythonImportsCheck = [ "optimistix" ];
47
+
48
+
nativeCheckInputs = [
49
+
beartype
50
+
jaxlib
51
+
optax
52
+
pytestCheckHook
53
+
pytest-xdist
54
+
];
55
+
56
+
meta = {
57
+
description = "Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox";
58
+
homepage = "https://github.com/patrick-kidger/optimistix";
59
+
changelog = "https://github.com/patrick-kidger/optimistix/releases/tag/v${version}";
60
+
license = lib.licenses.asl20;
61
+
maintainers = with lib.maintainers; [ GaetanLepage ];
62
+
};
63
+
}
+2
pkgs/top-level/python-packages.nix
+2
pkgs/top-level/python-packages.nix
···
9410
9410
9411
9411
optax = callPackage ../development/python-modules/optax { };
9412
9412
9413
+
optimistix = callPackage ../development/python-modules/optimistix { };
9414
+
9413
9415
optimum = callPackage ../development/python-modules/optimum { };
9414
9416
9415
9417
optree = callPackage ../development/python-modules/optree { };