···

       
1
       
 
       
{

     

       
0
       
 
       

     

       
0
       
 
       

     

       
2
       
 
       
  buildPythonPackage,

     

       
3
       
 
       
  fetchFromGitHub,

     

       
4
       
-
       
  lib,

     

       
5
       
 
       
  replaceVars,

     

       
6
       
-
       
  stdenv,

     

       
7
       
 
       


     

       
8
       
 
       
  # build-system

     

       
9
       
 
       
  setuptools,

     
···

       
13
       
 
       


     

       
14
       
 
       
  # buildInputs

     

       
15
       
 
       
  apple-sdk_14,

     

       
16
       
-
       
  fmt_10,

     

       
17
       
 
       
  nanobind,

     

       
18
       
 
       
  nlohmann_json,

     

       
19
       
 
       
  pybind11,

     
···

       
36
       
 
       


     

       
37
       
 
       
  mlx = buildPythonPackage rec {

     

       
38
       
 
       
    pname = "mlx";

     

       
39
       
-
       
    version = "0.26.3";

     

       
40
       
 
       
    pyproject = true;

     

       
41
       
 
       


     

       
42
       
 
       
    src = fetchFromGitHub {

     

       
43
       
 
       
      owner = "ml-explore";

     

       
44
       
 
       
      repo = "mlx";

     

       
45
       
 
       
      tag = "v${version}";

     

       
46
       
-
       
      hash = "sha256-hbqV/2KYGJ1gyExZd5bgaxTdhl5+Gext+U/+1KAztMU=";

     

       
47
       
 
       
    };

     

       
48
       
 
       


     

       
49
       
 
       
    patches = [

     
···

       
80
       
 
       
        # hatches which let you interact with a native install of Xcode, such as `composeXcodeWrapper`

     

       
81
       
 
       
        # or by changing the upstream (e.g., https://github.com/zed-industries/zed/discussions/7016).

     

       
82
       
 
       
        (lib.cmakeBool "MLX_BUILD_METAL" false)

     

       
0
       
 
       

     

       
83
       
 
       
        (lib.cmakeOptionType "filepath" "FETCHCONTENT_SOURCE_DIR_GGUFLIB" "${gguf-tools}")

     

       
84
       
 
       
        (lib.cmakeOptionType "filepath" "FETCHCONTENT_SOURCE_DIR_JSON" "${nlohmann_json.src}")

     

       
85
       
-
       
        (lib.cmakeOptionType "filepath" "FETCHCONTENT_SOURCE_DIR_FMT" "${fmt_10.src}")

     

       
86
       
 
       
      ];

     

       
87
       
 
       
    };

     

       
88
       
 
       


     
···

       
96
       
 
       


     

       
97
       
 
       
    buildInputs = [

     

       
98
       
 
       
      apple-sdk_14

     

       
99
       
-
       
      fmt_10

     

       
100
       
 
       
      gguf-tools

     

       
101
       
 
       
      nanobind

     

       
102
       
 
       
      nlohmann_json

     
···

       
113
       
 
       


     

       
114
       
 
       
    enabledTestPaths = [

     

       
115
       
 
       
      "python/tests/"

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
116
       
 
       
    ];

     

       
117
       
 
       


     

       
118
       
 
       
    # Additional testing by executing the example Python scripts supplied with mlx

     

···

       
1
       
 
       
{

     

       
2
       
+
       
  lib,

     

       
3
       
+
       
  stdenv,

     

       
4
       
 
       
  buildPythonPackage,

     

       
5
       
 
       
  fetchFromGitHub,

     

       
0
       
 
       

     

       
6
       
 
       
  replaceVars,

     

       
0
       
 
       

     

       
7
       
 
       


     

       
8
       
 
       
  # build-system

     

       
9
       
 
       
  setuptools,

     
···

       
13
       
 
       


     

       
14
       
 
       
  # buildInputs

     

       
15
       
 
       
  apple-sdk_14,

     

       
16
       
+
       
  fmt_11,

     

       
17
       
 
       
  nanobind,

     

       
18
       
 
       
  nlohmann_json,

     

       
19
       
 
       
  pybind11,

     
···

       
36
       
 
       


     

       
37
       
 
       
  mlx = buildPythonPackage rec {

     

       
38
       
 
       
    pname = "mlx";

     

       
39
       
+
       
    version = "0.28.0";

     

       
40
       
 
       
    pyproject = true;

     

       
41
       
 
       


     

       
42
       
 
       
    src = fetchFromGitHub {

     

       
43
       
 
       
      owner = "ml-explore";

     

       
44
       
 
       
      repo = "mlx";

     

       
45
       
 
       
      tag = "v${version}";

     

       
46
       
+
       
      hash = "sha256-+2dVZ89a09q8mWIbv6fBsySp7clzRV1tOyqr5hjFrNU=";

     

       
47
       
 
       
    };

     

       
48
       
 
       


     

       
49
       
 
       
    patches = [

     
···

       
80
       
 
       
        # hatches which let you interact with a native install of Xcode, such as `composeXcodeWrapper`

     

       
81
       
 
       
        # or by changing the upstream (e.g., https://github.com/zed-industries/zed/discussions/7016).

     

       
82
       
 
       
        (lib.cmakeBool "MLX_BUILD_METAL" false)

     

       
83
       
+
       
        (lib.cmakeBool "USE_SYSTEM_FMT" true)

     

       
84
       
 
       
        (lib.cmakeOptionType "filepath" "FETCHCONTENT_SOURCE_DIR_GGUFLIB" "${gguf-tools}")

     

       
85
       
 
       
        (lib.cmakeOptionType "filepath" "FETCHCONTENT_SOURCE_DIR_JSON" "${nlohmann_json.src}")

     

       
0
       
 
       

     

       
86
       
 
       
      ];

     

       
87
       
 
       
    };

     

       
88
       
 
       


     
···

       
96
       
 
       


     

       
97
       
 
       
    buildInputs = [

     

       
98
       
 
       
      apple-sdk_14

     

       
99
       
+
       
      fmt_11

     

       
100
       
 
       
      gguf-tools

     

       
101
       
 
       
      nanobind

     

       
102
       
 
       
      nlohmann_json

     
···

       
113
       
 
       


     

       
114
       
 
       
    enabledTestPaths = [

     

       
115
       
 
       
      "python/tests/"

     

       
116
       
+
       
    ];

     

       
117
       
+
       


     

       
118
       
+
       
    disabledTests = [

     

       
119
       
+
       
      # AssertionError

     

       
120
       
+
       
      "test_numpy_conv"

     

       
121
       
+
       
      "test_tensordot"

     

       
122
       
+
       
    ];

     

       
123
       
+
       


     

       
124
       
+
       
    disabledTestPaths = [

     

       
125
       
+
       
      # AssertionError

     

       
126
       
+
       
      "python/tests/test_blas.py"

     

       
127
       
 
       
    ];

     

       
128
       
 
       


     

       
129
       
 
       
    # Additional testing by executing the example Python scripts supplied with mlx