tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 torch,
6}:
7
8buildPythonPackage rec {
9 pname = "lion-pytorch";
10 version = "0.2.3";
11 format = "setuptools";
12
13 src = fetchFromGitHub {
14 owner = "lucidrains";
15 repo = "lion-pytorch";
16 tag = version;
17 hash = "sha256-RHixPIZ1kCawWQiqYqLY+c3r6Rg86LKm3tQTyW2BNFU=";
18 };
19
20 propagatedBuildInputs = [ torch ];
21
22 pythonImportsCheck = [ "lion_pytorch" ];
23 doCheck = false; # no tests currently
24
25 meta = {
26 description = "Optimizer tuned by Google Brain using genetic algorithms";
27 homepage = "https://github.com/lucidrains/lion-pytorch";
28 license = lib.licenses.mit;
29 maintainers = with lib.maintainers; [ bcdarwin ];
30 };
31}