···

       
1
       
1
       
 
       
{

     

       
2
       
2
       
 
       
  lib,

     

       
3
       
3
       
 
       
  buildPythonPackage,

     

       
4
       
4
       
+
       
  pythonAtLeast,

     

       
4
       
5
       
 
       
  fetchFromGitHub,

     

       
5
       
6
       
 
       
  fetchpatch,

     

       
7
       
7
       
+
       


     

       
8
       
8
       
+
       
  # build-system

     

       
6
       
9
       
 
       
  setuptools,

     

       
10
       
10
       
+
       


     

       
11
       
11
       
+
       
  # dependencies

     

       
7
       
12
       
 
       
  absl-py,

     

       
8
       
8
       
-
       
  flax,

     

       
9
       
9
       
-
       
  jax,

     

       
10
       
13
       
 
       
  jaxlib,

     

       
11
       
14
       
 
       
  jmp,

     

       
12
       
15
       
 
       
  numpy,

     

       
13
       
16
       
 
       
  tabulate,

     

       
17
       
17
       
+
       


     

       
18
       
18
       
+
       
  # optional-dependencies

     

       
19
       
19
       
+
       
  jax,

     

       
20
       
20
       
+
       
  flax,

     

       
21
       
21
       
+
       


     

       
22
       
22
       
+
       
  # tests

     

       
14
       
23
       
 
       
  pytest-xdist,

     

       
15
       
24
       
 
       
  pytestCheckHook,

     

       
16
       
25
       
 
       
  bsuite,

     
···

       
29
       
38
       
 
       
    pname = "dm-haiku";

     

       
30
       
39
       
 
       
    version = "0.0.13";

     

       
31
       
40
       
 
       
    pyproject = true;

     

       
41
       
41
       
+
       


     

       
42
       
42
       
+
       
    # ImportError: `haiku.experimental.flax` features require `flax` to be installed.

     

       
43
       
43
       
+
       
    disabled = pythonAtLeast "3.13";

     

       
32
       
44
       
 
       


     

       
33
       
45
       
 
       
    src = fetchFromGitHub {

     

       
34
       
46
       
 
       
      owner = "deepmind";

     
···

       
117
       
129
       
 
       
      dontInstall = true;

     

       
118
       
130
       
 
       
    });

     

       
119
       
131
       
 
       


     

       
120
       
120
       
-
       
    meta = with lib; {

     

       
132
       
132
       
+
       
    meta = {

     

       
121
       
133
       
 
       
      description = "Haiku is a simple neural network library for JAX developed by some of the authors of Sonnet";

     

       
122
       
134
       
 
       
      homepage = "https://github.com/deepmind/dm-haiku";

     

       
123
       
123
       
-
       
      license = licenses.asl20;

     

       
124
       
124
       
-
       
      maintainers = with maintainers; [ ndl ];

     

       
135
       
135
       
+
       
      license = lib.licenses.asl20;

     

       
136
       
136
       
+
       
      maintainers = with lib.maintainers; [ ndl ];

     

       
125
       
137
       
 
       
    };

     

       
126
       
138
       
 
       
  };

     

       
127
       
139
       
 
       
in