python310Packages.xformers: init at 0.0.20

{ lib

, buildPythonPackage

, pythonOlder

, fetchFromGitHub

, pythonRelaxDepsHook

, which

# runtime dependencies

, numpy

, torch

, pyre-extensions

# check dependencies

, pytestCheckHook

, pytest-cov

# , pytest-mpi

, pytest-timeout

# , pytorch-image-models

, hydra-core

, fairscale

, scipy

, cmake

, openai-triton

, networkx

}:

let

  version = "0.0.20";

in

buildPythonPackage {

  pname = "xformers";

  inherit version;

  format = "setuptools";

  disable = pythonOlder "3.7";

  src = fetchFromGitHub {

    owner = "facebookresearch";

    repo = "xformers";

    rev = "v${version}";

    hash = "sha256-OFH4I3eTKw1bQEKHh1AvkpcoShKK5R5674AoJ/mY85I=";

    fetchSubmodules = true;

  };

  preBuild = ''

    cat << EOF > ./xformers/version.py

    # noqa: C801

    __version__ = "${version}"

    EOF

  '';

  nativeBuildInputs = [

    pythonRelaxDepsHook

    which

  ];

  pythonRelaxDeps = [

    "pyre-extensions"

  ];

  propagatedBuildInputs = [

    numpy

    torch

    pyre-extensions

  ];

  pythonImportsCheck = [ "xformers" ];

  nativeCheckInputs = [

    pytestCheckHook

    pytest-cov

    pytest-timeout

    hydra-core

    fairscale

    scipy

    cmake

    networkx

    openai-triton

  ];

  meta = with lib; {

    description = "XFormers: A collection of composable Transformer building blocks";

    homepage = "https://github.com/facebookresearch/xformers";

    changelog = "https://github.com/facebookresearch/xformers/blob/${version}/CHANGELOG.md";

    license = licenses.bsd3;

    maintainers = with maintainers; [ happysalada ];

  };

}

    inherit (pkgs) graphviz;

  };

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

  xgboost = callPackage ../development/python-modules/xgboost {

    inherit (pkgs) xgboost;

  };