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