python312Packages.flowjax: init at 17.1.1

{

  lib,

  buildPythonPackage,

  fetchFromGitHub,

  # build-system

  hatchling,

  # dependencies

  equinox,

  jax,

  jaxtyping,

  optax,

  paramax,

  tqdm,

  # tests

  beartype,

  numpyro,

  pytest-xdist,

  pytestCheckHook,

}:

buildPythonPackage rec {

  pname = "flowjax";

  version = "17.1.1";

  pyproject = true;

  src = fetchFromGitHub {

    owner = "danielward27";

    repo = "flowjax";

    tag = "v${version}";

    hash = "sha256-CLtkO7Lr+FpC/RHnqDSmevpJ3/3EfuILrQSbcdNTqsI=";

  };

  build-system = [

    hatchling

  ];

  dependencies = [

    equinox

    jax

    jaxtyping

    optax

    paramax

    tqdm

  ];

  pythonImportsCheck = [ "flowjax" ];

  nativeCheckInputs = [

    beartype

    numpyro

    pytest-xdist

    pytestCheckHook

  ];

  meta = {

    description = "Distributions, bijections and normalizing flows using Equinox and JAX";

    homepage = "https://github.com/danielward27/flowjax";

    changelog = "https://github.com/danielward27/flowjax/releases/tag/v${version}";

    license = lib.licenses.mit;

    maintainers = with lib.maintainers; [ GaetanLepage ];

  };

}

  flower = callPackage ../development/python-modules/flower { };

  flowjax = callPackage ../development/python-modules/flowjax { };

  flowlogs-reader = callPackage ../development/python-modules/flowlogs-reader { };

  flowmc = callPackage ../development/python-modules/flowmc { };