diff --git a/pkgs/development/python-modules/optimistix/default.nix b/pkgs/development/python-modules/optimistix/default.nix new file mode 100644 index 000000000000000..a7cb8b7c5e7e687 --- /dev/null +++ b/pkgs/development/python-modules/optimistix/default.nix @@ -0,0 +1,63 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + + # build-system + hatchling, + + # dependencies + equinox, + jax, + jaxtyping, + lineax, + typing-extensions, + + # checks + beartype, + jaxlib, + optax, + pytestCheckHook, + pytest-xdist, +}: + +buildPythonPackage rec { + pname = "optimistix"; + version = "0.0.8"; + pyproject = true; + + src = fetchFromGitHub { + owner = "patrick-kidger"; + repo = "optimistix"; + rev = "refs/tags/v${version}"; + hash = "sha256-0ehNApiBoiAb8LFBW81ZCRPsjTVQG8zRTVSAp7oHF8w="; + }; + + build-system = [ hatchling ]; + + dependencies = [ + equinox + jax + jaxtyping + lineax + typing-extensions + ]; + + pythonImportsCheck = [ "optimistix" ]; + + nativeCheckInputs = [ + beartype + jaxlib + optax + pytestCheckHook + pytest-xdist + ]; + + meta = { + description = "Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox"; + homepage = "https://github.com/patrick-kidger/optimistix"; + changelog = "https://github.com/patrick-kidger/optimistix/releases/tag/v${version}"; + license = lib.licenses.asl20; + maintainers = with lib.maintainers; [ GaetanLepage ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 7168fe5f4d0addb..6a2dbc777c089dd 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -9410,6 +9410,8 @@ self: super: with self; { optax = callPackage ../development/python-modules/optax { }; + optimistix = callPackage ../development/python-modules/optimistix { }; + optimum = callPackage ../development/python-modules/optimum { }; optree = callPackage ../development/python-modules/optree { };