quabla 0.6.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (107) hide show
  1. quabla-0.6.1/Cargo.lock +820 -0
  2. quabla-0.6.1/Cargo.toml +15 -0
  3. quabla-0.6.1/LICENSE-APACHE +202 -0
  4. quabla-0.6.1/LICENSE-MIT +21 -0
  5. quabla-0.6.1/PKG-INFO +416 -0
  6. quabla-0.6.1/README.md +383 -0
  7. quabla-0.6.1/crates/quabla-core/Cargo.toml +33 -0
  8. quabla-0.6.1/crates/quabla-core/examples/fit_lotka_volterra.rs +36 -0
  9. quabla-0.6.1/crates/quabla-core/examples/lotka_volterra.rs +23 -0
  10. quabla-0.6.1/crates/quabla-core/src/autodiff.rs +192 -0
  11. quabla-0.6.1/crates/quabla-core/src/compiler.rs +483 -0
  12. quabla-0.6.1/crates/quabla-core/src/cpu_kernel_indexing_tests.rs +307 -0
  13. quabla-0.6.1/crates/quabla-core/src/lib.rs +23 -0
  14. quabla-0.6.1/crates/quabla-core/src/models.rs +100 -0
  15. quabla-0.6.1/crates/quabla-core/src/ode.rs +114 -0
  16. quabla-0.6.1/crates/quabla-core/src/optim.rs +65 -0
  17. quabla-0.6.1/crates/quabla-core/src/solve_replay_tests.rs +338 -0
  18. quabla-0.6.1/crates/quabla-core/src/tensor.rs +67 -0
  19. quabla-0.6.1/crates/quabla-core/src/tensor_ir/cholesky.rs +277 -0
  20. quabla-0.6.1/crates/quabla-core/src/tensor_ir/cholesky_tests.rs +464 -0
  21. quabla-0.6.1/crates/quabla-core/src/tensor_ir/cuda.rs +9915 -0
  22. quabla-0.6.1/crates/quabla-core/src/tensor_ir/cuda_cholesky.rs +188 -0
  23. quabla-0.6.1/crates/quabla-core/src/tensor_ir/cuda_decompositions.rs +545 -0
  24. quabla-0.6.1/crates/quabla-core/src/tensor_ir/cuda_host_loop.rs +2174 -0
  25. quabla-0.6.1/crates/quabla-core/src/tensor_ir/cuda_optimizer.rs +547 -0
  26. quabla-0.6.1/crates/quabla-core/src/tensor_ir/cuda_real.rs +635 -0
  27. quabla-0.6.1/crates/quabla-core/src/tensor_ir/custom.rs +770 -0
  28. quabla-0.6.1/crates/quabla-core/src/tensor_ir/device_optimizer.rs +775 -0
  29. quabla-0.6.1/crates/quabla-core/src/tensor_ir/eager.rs +850 -0
  30. quabla-0.6.1/crates/quabla-core/src/tensor_ir/elementwise.rs +1365 -0
  31. quabla-0.6.1/crates/quabla-core/src/tensor_ir/extremum.rs +225 -0
  32. quabla-0.6.1/crates/quabla-core/src/tensor_ir/fusion.rs +2070 -0
  33. quabla-0.6.1/crates/quabla-core/src/tensor_ir/host_storage.rs +245 -0
  34. quabla-0.6.1/crates/quabla-core/src/tensor_ir/linalg.rs +636 -0
  35. quabla-0.6.1/crates/quabla-core/src/tensor_ir/mlx.rs +2711 -0
  36. quabla-0.6.1/crates/quabla-core/src/tensor_ir/mlx_checkpoint_tests.rs +332 -0
  37. quabla-0.6.1/crates/quabla-core/src/tensor_ir/mlx_cholesky.rs +357 -0
  38. quabla-0.6.1/crates/quabla-core/src/tensor_ir/mlx_cholesky_tests.rs +254 -0
  39. quabla-0.6.1/crates/quabla-core/src/tensor_ir/mlx_linalg.rs +534 -0
  40. quabla-0.6.1/crates/quabla-core/src/tensor_ir/mlx_scatter.rs +192 -0
  41. quabla-0.6.1/crates/quabla-core/src/tensor_ir/region.rs +413 -0
  42. quabla-0.6.1/crates/quabla-core/src/tensor_ir/region_batching.rs +1038 -0
  43. quabla-0.6.1/crates/quabla-core/src/tensor_ir/region_batching_tests.rs +558 -0
  44. quabla-0.6.1/crates/quabla-core/src/tensor_ir.rs +19061 -0
  45. quabla-0.6.1/crates/quabla-core/tests/cholesky_memory.rs +206 -0
  46. quabla-0.6.1/crates/quabla-core/tests/compiler_liveness.rs +373 -0
  47. quabla-0.6.1/crates/quabla-core/tests/cpu_execution_memory.rs +250 -0
  48. quabla-0.6.1/crates/quabla-core/tests/cpu_loop_checkpoint.rs +459 -0
  49. quabla-0.6.1/crates/quabla-core/tests/cuda_axis_reduction.rs +131 -0
  50. quabla-0.6.1/crates/quabla-core/tests/cuda_codegen_memory.rs +93 -0
  51. quabla-0.6.1/crates/quabla-core/tests/cuda_float64.rs +226 -0
  52. quabla-0.6.1/crates/quabla-core/tests/cuda_native_cholesky.rs +414 -0
  53. quabla-0.6.1/crates/quabla-core/tests/cuda_retained_loss.rs +151 -0
  54. quabla-0.6.1/crates/quabla-core/tests/cuda_solve_profile.rs +49 -0
  55. quabla-0.6.1/crates/quabla-core/tests/direct_solve_jvp.rs +185 -0
  56. quabla-0.6.1/crates/quabla-core/tests/dual.rs +70 -0
  57. quabla-0.6.1/crates/quabla-core/tests/higher_order_optimization.rs +194 -0
  58. quabla-0.6.1/crates/quabla-core/tests/ode.rs +274 -0
  59. quabla-0.6.1/crates/quabla-core/tests/optim.rs +57 -0
  60. quabla-0.6.1/crates/quabla-core/tests/scan_numeric_jvp.rs +449 -0
  61. quabla-0.6.1/crates/quabla-core/tests/support/mod.rs +33 -0
  62. quabla-0.6.1/crates/quabla-core/tests/tensor.rs +59 -0
  63. quabla-0.6.1/crates/quabla-core/tests/tensor_ir.rs +14380 -0
  64. quabla-0.6.1/crates/quabla-core/tests/tensor_ir_linalg.rs +753 -0
  65. quabla-0.6.1/crates/quabla-macros/Cargo.toml +17 -0
  66. quabla-0.6.1/crates/quabla-macros/src/lib.rs +139 -0
  67. quabla-0.6.1/crates/quabla-python/Cargo.toml +24 -0
  68. quabla-0.6.1/crates/quabla-python/src/compiler.rs +378 -0
  69. quabla-0.6.1/crates/quabla-python/src/composite.rs +141 -0
  70. quabla-0.6.1/crates/quabla-python/src/dtype.rs +50 -0
  71. quabla-0.6.1/crates/quabla-python/src/errors.rs +87 -0
  72. quabla-0.6.1/crates/quabla-python/src/interop.rs +547 -0
  73. quabla-0.6.1/crates/quabla-python/src/lib.rs +560 -0
  74. quabla-0.6.1/crates/quabla-python/src/matrix.rs +743 -0
  75. quabla-0.6.1/crates/quabla-python/src/optim.rs +179 -0
  76. quabla-0.6.1/crates/quabla-python/src/repr.rs +410 -0
  77. quabla-0.6.1/crates/quabla-python/src/tensor.rs +3143 -0
  78. quabla-0.6.1/crates/quabla-python/src/tensor_trace.rs +9632 -0
  79. quabla-0.6.1/crates/quabla-python/src/trace.rs +4311 -0
  80. quabla-0.6.1/crates/quabla-python/tests/matrix.rs +217 -0
  81. quabla-0.6.1/crates/quabla-python/tests/tensor_trace.rs +277 -0
  82. quabla-0.6.1/crates/quabla-python/tests/trace.rs +621 -0
  83. quabla-0.6.1/pyproject.toml +62 -0
  84. quabla-0.6.1/python/quabla/__init__.py +122 -0
  85. quabla-0.6.1/python/quabla/_array.py +127 -0
  86. quabla-0.6.1/python/quabla/_compat.py +140 -0
  87. quabla-0.6.1/python/quabla/_control.py +269 -0
  88. quabla-0.6.1/python/quabla/_custom.py +675 -0
  89. quabla-0.6.1/python/quabla/_devices.py +50 -0
  90. quabla-0.6.1/python/quabla/_errors.py +49 -0
  91. quabla-0.6.1/python/quabla/_implicit.py +217 -0
  92. quabla-0.6.1/python/quabla/_io.py +243 -0
  93. quabla-0.6.1/python/quabla/_krylov.py +488 -0
  94. quabla-0.6.1/python/quabla/_lbfgs.py +327 -0
  95. quabla-0.6.1/python/quabla/_newton.py +252 -0
  96. quabla-0.6.1/python/quabla/_ops.py +1281 -0
  97. quabla-0.6.1/python/quabla/_quabla.pyi +722 -0
  98. quabla-0.6.1/python/quabla/_schedules.py +151 -0
  99. quabla-0.6.1/python/quabla/_transforms.py +2166 -0
  100. quabla-0.6.1/python/quabla/distributed.py +197 -0
  101. quabla-0.6.1/python/quabla/legacy.py +59 -0
  102. quabla-0.6.1/python/quabla/linalg.py +524 -0
  103. quabla-0.6.1/python/quabla/ode.py +590 -0
  104. quabla-0.6.1/python/quabla/optim.py +494 -0
  105. quabla-0.6.1/python/quabla/py.typed +0 -0
  106. quabla-0.6.1/python/quabla/random.py +203 -0
  107. quabla-0.6.1/python/quabla/tree.py +332 -0
@@ -0,0 +1,820 @@
1
+ # This file is automatically @generated by Cargo.
2
+ # It is not intended for manual editing.
3
+ version = 4
4
+
5
+ [[package]]
6
+ name = "aho-corasick"
7
+ version = "1.1.4"
8
+ source = "registry+https://github.com/rust-lang/crates.io-index"
9
+ checksum = "ddd31a130427c27518df266943a5308ed92d4b226cc639f5a8f1002816174301"
10
+ dependencies = [
11
+ "memchr",
12
+ ]
13
+
14
+ [[package]]
15
+ name = "autocfg"
16
+ version = "1.5.1"
17
+ source = "registry+https://github.com/rust-lang/crates.io-index"
18
+ checksum = "f2032f911046de80f0a198e0901378627c33f59ea0ac00e363d481118bd70a53"
19
+
20
+ [[package]]
21
+ name = "bindgen"
22
+ version = "0.72.1"
23
+ source = "registry+https://github.com/rust-lang/crates.io-index"
24
+ checksum = "993776b509cfb49c750f11b8f07a46fa23e0a1386ffc01fb1e7d343efc387895"
25
+ dependencies = [
26
+ "bitflags",
27
+ "cexpr",
28
+ "clang-sys",
29
+ "itertools 0.13.0",
30
+ "log",
31
+ "prettyplease",
32
+ "proc-macro2",
33
+ "quote",
34
+ "regex",
35
+ "rustc-hash",
36
+ "shlex 1.3.0",
37
+ "syn 2.0.119",
38
+ ]
39
+
40
+ [[package]]
41
+ name = "bitflags"
42
+ version = "2.13.0"
43
+ source = "registry+https://github.com/rust-lang/crates.io-index"
44
+ checksum = "b4388bee8683e3d04af747c73422af53102d2bd24d9eadb6cbc100baef4b43f8"
45
+
46
+ [[package]]
47
+ name = "cc"
48
+ version = "1.2.67"
49
+ source = "registry+https://github.com/rust-lang/crates.io-index"
50
+ checksum = "e17dd265a7d0f31ef544e1b20e03add05d3b45b491b633b10d67145d2acc1a38"
51
+ dependencies = [
52
+ "find-msvc-tools",
53
+ "shlex 2.0.1",
54
+ ]
55
+
56
+ [[package]]
57
+ name = "cexpr"
58
+ version = "0.6.0"
59
+ source = "registry+https://github.com/rust-lang/crates.io-index"
60
+ checksum = "6fac387a98bb7c37292057cffc56d62ecb629900026402633ae9160df93a8766"
61
+ dependencies = [
62
+ "nom",
63
+ ]
64
+
65
+ [[package]]
66
+ name = "cfg-if"
67
+ version = "1.0.4"
68
+ source = "registry+https://github.com/rust-lang/crates.io-index"
69
+ checksum = "9330f8b2ff13f34540b44e946ef35111825727b38d33286ef986142615121801"
70
+
71
+ [[package]]
72
+ name = "clang-sys"
73
+ version = "1.8.1"
74
+ source = "registry+https://github.com/rust-lang/crates.io-index"
75
+ checksum = "0b023947811758c97c59bf9d1c188fd619ad4718dcaa767947df1cadb14f39f4"
76
+ dependencies = [
77
+ "glob",
78
+ "libc",
79
+ "libloading 0.8.9",
80
+ ]
81
+
82
+ [[package]]
83
+ name = "cmake"
84
+ version = "0.1.58"
85
+ source = "registry+https://github.com/rust-lang/crates.io-index"
86
+ checksum = "c0f78a02292a74a88ac736019ab962ece0bc380e3f977bf72e376c5d78ff0678"
87
+ dependencies = [
88
+ "cc",
89
+ ]
90
+
91
+ [[package]]
92
+ name = "crunchy"
93
+ version = "0.2.4"
94
+ source = "registry+https://github.com/rust-lang/crates.io-index"
95
+ checksum = "460fbee9c2c2f33933d720630a6a0bac33ba7053db5344fac858d4b8952d77d5"
96
+
97
+ [[package]]
98
+ name = "cudarc"
99
+ version = "0.19.10"
100
+ source = "registry+https://github.com/rust-lang/crates.io-index"
101
+ checksum = "7359bf1de037ddada72729c3acec9b040f3468400c6e9e740b01ba0cd2e89562"
102
+ dependencies = [
103
+ "libloading 0.9.0",
104
+ ]
105
+
106
+ [[package]]
107
+ name = "darling"
108
+ version = "0.21.3"
109
+ source = "registry+https://github.com/rust-lang/crates.io-index"
110
+ checksum = "9cdf337090841a411e2a7f3deb9187445851f91b309c0c0a29e05f74a00a48c0"
111
+ dependencies = [
112
+ "darling_core",
113
+ "darling_macro",
114
+ ]
115
+
116
+ [[package]]
117
+ name = "darling_core"
118
+ version = "0.21.3"
119
+ source = "registry+https://github.com/rust-lang/crates.io-index"
120
+ checksum = "1247195ecd7e3c85f83c8d2a366e4210d588e802133e1e355180a9870b517ea4"
121
+ dependencies = [
122
+ "fnv",
123
+ "ident_case",
124
+ "proc-macro2",
125
+ "quote",
126
+ "strsim",
127
+ "syn 2.0.119",
128
+ ]
129
+
130
+ [[package]]
131
+ name = "darling_macro"
132
+ version = "0.21.3"
133
+ source = "registry+https://github.com/rust-lang/crates.io-index"
134
+ checksum = "d38308df82d1080de0afee5d069fa14b0326a88c14f15c5ccda35b4a6c414c81"
135
+ dependencies = [
136
+ "darling_core",
137
+ "quote",
138
+ "syn 2.0.119",
139
+ ]
140
+
141
+ [[package]]
142
+ name = "dyn-clone"
143
+ version = "1.0.20"
144
+ source = "registry+https://github.com/rust-lang/crates.io-index"
145
+ checksum = "d0881ea181b1df73ff77ffaaf9c7544ecc11e82fba9b5f27b262a3c73a332555"
146
+
147
+ [[package]]
148
+ name = "either"
149
+ version = "1.16.0"
150
+ source = "registry+https://github.com/rust-lang/crates.io-index"
151
+ checksum = "91622ff5e7162018101f2fea40d6ebf4a78bbe5a49736a2020649edf9693679e"
152
+
153
+ [[package]]
154
+ name = "equivalent"
155
+ version = "1.0.2"
156
+ source = "registry+https://github.com/rust-lang/crates.io-index"
157
+ checksum = "877a4ace8713b0bcf2a4e7eec82529c029f1d0619886d18145fea96c3ffe5c0f"
158
+
159
+ [[package]]
160
+ name = "fastrand"
161
+ version = "2.4.1"
162
+ source = "registry+https://github.com/rust-lang/crates.io-index"
163
+ checksum = "9f1f227452a390804cdb637b74a86990f2a7d7ba4b7d5693aac9b4dd6defd8d6"
164
+
165
+ [[package]]
166
+ name = "find-msvc-tools"
167
+ version = "0.1.9"
168
+ source = "registry+https://github.com/rust-lang/crates.io-index"
169
+ checksum = "5baebc0774151f905a1a2cc41989300b1e6fbb29aff0ceffa1064fdd3088d582"
170
+
171
+ [[package]]
172
+ name = "fnv"
173
+ version = "1.0.7"
174
+ source = "registry+https://github.com/rust-lang/crates.io-index"
175
+ checksum = "3f9eec918d3f24069decb9af1554cad7c880e2da24a9afd88aca000531ab82c1"
176
+
177
+ [[package]]
178
+ name = "glob"
179
+ version = "0.3.3"
180
+ source = "registry+https://github.com/rust-lang/crates.io-index"
181
+ checksum = "0cc23270f6e1808e30a928bdc84dea0b9b4136a8bc82338574f23baf47bbd280"
182
+
183
+ [[package]]
184
+ name = "half"
185
+ version = "2.7.1"
186
+ source = "registry+https://github.com/rust-lang/crates.io-index"
187
+ checksum = "6ea2d84b969582b4b1864a92dc5d27cd2b77b622a8d79306834f1be5ba20d84b"
188
+ dependencies = [
189
+ "cfg-if",
190
+ "crunchy",
191
+ "zerocopy",
192
+ ]
193
+
194
+ [[package]]
195
+ name = "hashbrown"
196
+ version = "0.17.1"
197
+ source = "registry+https://github.com/rust-lang/crates.io-index"
198
+ checksum = "ed5909b6e89a2db4456e54cd5f673791d7eca6732202bbf2a9cc504fe2f9b84a"
199
+
200
+ [[package]]
201
+ name = "heck"
202
+ version = "0.5.0"
203
+ source = "registry+https://github.com/rust-lang/crates.io-index"
204
+ checksum = "2304e00983f87ffb38b55b444b5e3b60a884b5d30c0fca7d82fe33449bbe55ea"
205
+
206
+ [[package]]
207
+ name = "ident_case"
208
+ version = "1.0.1"
209
+ source = "registry+https://github.com/rust-lang/crates.io-index"
210
+ checksum = "b9e0384b61958566e926dc50660321d12159025e767c18e043daf26b70104c39"
211
+
212
+ [[package]]
213
+ name = "indexmap"
214
+ version = "2.14.0"
215
+ source = "registry+https://github.com/rust-lang/crates.io-index"
216
+ checksum = "d466e9454f08e4a911e14806c24e16fba1b4c121d1ea474396f396069cf949d9"
217
+ dependencies = [
218
+ "equivalent",
219
+ "hashbrown",
220
+ ]
221
+
222
+ [[package]]
223
+ name = "itertools"
224
+ version = "0.13.0"
225
+ source = "registry+https://github.com/rust-lang/crates.io-index"
226
+ checksum = "413ee7dfc52ee1a4949ceeb7dbc8a33f2d6c088194d9f922fb8318faf1f01186"
227
+ dependencies = [
228
+ "either",
229
+ ]
230
+
231
+ [[package]]
232
+ name = "itertools"
233
+ version = "0.14.0"
234
+ source = "registry+https://github.com/rust-lang/crates.io-index"
235
+ checksum = "2b192c782037fadd9cfa75548310488aabdbf3d2da73885b31bd0abd03351285"
236
+ dependencies = [
237
+ "either",
238
+ ]
239
+
240
+ [[package]]
241
+ name = "libc"
242
+ version = "0.2.186"
243
+ source = "registry+https://github.com/rust-lang/crates.io-index"
244
+ checksum = "68ab91017fe16c622486840e4c83c9a37afeff978bd239b5293d61ece587de66"
245
+
246
+ [[package]]
247
+ name = "libloading"
248
+ version = "0.8.9"
249
+ source = "registry+https://github.com/rust-lang/crates.io-index"
250
+ checksum = "d7c4b02199fee7c5d21a5ae7d8cfa79a6ef5bb2fc834d6e9058e89c825efdc55"
251
+ dependencies = [
252
+ "cfg-if",
253
+ "windows-link",
254
+ ]
255
+
256
+ [[package]]
257
+ name = "libloading"
258
+ version = "0.9.0"
259
+ source = "registry+https://github.com/rust-lang/crates.io-index"
260
+ checksum = "754ca22de805bb5744484a5b151a9e1a8e837d5dc232c2d7d8c2e3492edc8b60"
261
+ dependencies = [
262
+ "cfg-if",
263
+ "windows-link",
264
+ ]
265
+
266
+ [[package]]
267
+ name = "libm"
268
+ version = "0.2.16"
269
+ source = "registry+https://github.com/rust-lang/crates.io-index"
270
+ checksum = "b6d2cec3eae94f9f509c767b45932f1ada8350c4bdb85af2fcab4a3c14807981"
271
+
272
+ [[package]]
273
+ name = "lock_api"
274
+ version = "0.4.14"
275
+ source = "registry+https://github.com/rust-lang/crates.io-index"
276
+ checksum = "224399e74b87b5f3557511d98dff8b14089b3dadafcab6bb93eab67d3aace965"
277
+ dependencies = [
278
+ "scopeguard",
279
+ ]
280
+
281
+ [[package]]
282
+ name = "log"
283
+ version = "0.4.33"
284
+ source = "registry+https://github.com/rust-lang/crates.io-index"
285
+ checksum = "0ceec5bc11778974d1bcb055b18002eba7f4b3518b6a0081b3af5f21666da9ad"
286
+
287
+ [[package]]
288
+ name = "mach-sys"
289
+ version = "0.5.4"
290
+ source = "registry+https://github.com/rust-lang/crates.io-index"
291
+ checksum = "48460c2e82a3a0de197152fdf8d2c2d5e43adc501501553e439bf2156e6f87c7"
292
+ dependencies = [
293
+ "fastrand",
294
+ ]
295
+
296
+ [[package]]
297
+ name = "memchr"
298
+ version = "2.8.3"
299
+ source = "registry+https://github.com/rust-lang/crates.io-index"
300
+ checksum = "cf8baf1c55e62ffcace7a9f06f4bd9cd3f0c4beb022d3b367256b91b87513d98"
301
+
302
+ [[package]]
303
+ name = "minimal-lexical"
304
+ version = "0.2.1"
305
+ source = "registry+https://github.com/rust-lang/crates.io-index"
306
+ checksum = "68354c5c6bd36d73ff3feceb05efa59b6acb7626617f4962be322a825e61f79a"
307
+
308
+ [[package]]
309
+ name = "mlx-internal-macros"
310
+ version = "0.32.0"
311
+ source = "registry+https://github.com/rust-lang/crates.io-index"
312
+ checksum = "bf724ff93a5ddcf64299f7786844301b147e0e37a7de30a7e1536fd60fb7e9f1"
313
+ dependencies = [
314
+ "darling",
315
+ "itertools 0.14.0",
316
+ "proc-macro2",
317
+ "quote",
318
+ "syn 2.0.119",
319
+ ]
320
+
321
+ [[package]]
322
+ name = "mlx-macros"
323
+ version = "0.32.0"
324
+ source = "registry+https://github.com/rust-lang/crates.io-index"
325
+ checksum = "df160e2aa7cd71a8bf652eeefa9a03117c775327071b68cb7750e1be7d6985e7"
326
+ dependencies = [
327
+ "darling",
328
+ "proc-macro2",
329
+ "quote",
330
+ "syn 2.0.119",
331
+ ]
332
+
333
+ [[package]]
334
+ name = "mlx-rs"
335
+ version = "0.32.0"
336
+ source = "registry+https://github.com/rust-lang/crates.io-index"
337
+ checksum = "44774de28d595d6bc0f9ae359e5365321b96eab252698ec0773406c0cf77762b"
338
+ dependencies = [
339
+ "dyn-clone",
340
+ "half",
341
+ "itertools 0.14.0",
342
+ "libc",
343
+ "mach-sys",
344
+ "mlx-internal-macros",
345
+ "mlx-macros",
346
+ "mlx-sys",
347
+ "num-complex",
348
+ "num-traits",
349
+ "num_enum",
350
+ "parking_lot",
351
+ "paste",
352
+ "smallvec",
353
+ "strum",
354
+ "thiserror",
355
+ ]
356
+
357
+ [[package]]
358
+ name = "mlx-sys"
359
+ version = "0.6.0"
360
+ source = "registry+https://github.com/rust-lang/crates.io-index"
361
+ checksum = "1848b6f1686114039e6e0adcf1578911e2e4d7707b9768ad61da074132319e1c"
362
+ dependencies = [
363
+ "bindgen",
364
+ "cc",
365
+ "cmake",
366
+ ]
367
+
368
+ [[package]]
369
+ name = "nom"
370
+ version = "7.1.3"
371
+ source = "registry+https://github.com/rust-lang/crates.io-index"
372
+ checksum = "d273983c5a657a70a3e8f2a01329822f3b8c8172b73826411a55751e404a0a4a"
373
+ dependencies = [
374
+ "memchr",
375
+ "minimal-lexical",
376
+ ]
377
+
378
+ [[package]]
379
+ name = "num-complex"
380
+ version = "0.4.6"
381
+ source = "registry+https://github.com/rust-lang/crates.io-index"
382
+ checksum = "73f88a1307638156682bada9d7604135552957b7818057dcef22705b4d509495"
383
+ dependencies = [
384
+ "num-traits",
385
+ ]
386
+
387
+ [[package]]
388
+ name = "num-traits"
389
+ version = "0.2.19"
390
+ source = "registry+https://github.com/rust-lang/crates.io-index"
391
+ checksum = "071dfc062690e90b734c0b2273ce72ad0ffa95f0c74596bc250dcfd960262841"
392
+ dependencies = [
393
+ "autocfg",
394
+ ]
395
+
396
+ [[package]]
397
+ name = "num_enum"
398
+ version = "0.7.6"
399
+ source = "registry+https://github.com/rust-lang/crates.io-index"
400
+ checksum = "5d0bca838442ec211fa11de3a8b0e0e8f3a4522575b5c4c06ed722e005036f26"
401
+ dependencies = [
402
+ "num_enum_derive",
403
+ "rustversion",
404
+ ]
405
+
406
+ [[package]]
407
+ name = "num_enum_derive"
408
+ version = "0.7.6"
409
+ source = "registry+https://github.com/rust-lang/crates.io-index"
410
+ checksum = "680998035259dcfcafe653688bf2aa6d3e2dc05e98be6ab46afb089dc84f1df8"
411
+ dependencies = [
412
+ "proc-macro-crate",
413
+ "proc-macro2",
414
+ "quote",
415
+ "syn 2.0.119",
416
+ ]
417
+
418
+ [[package]]
419
+ name = "once_cell"
420
+ version = "1.21.4"
421
+ source = "registry+https://github.com/rust-lang/crates.io-index"
422
+ checksum = "9f7c3e4beb33f85d45ae3e3a1792185706c8e16d043238c593331cc7cd313b50"
423
+
424
+ [[package]]
425
+ name = "parking_lot"
426
+ version = "0.12.5"
427
+ source = "registry+https://github.com/rust-lang/crates.io-index"
428
+ checksum = "93857453250e3077bd71ff98b6a65ea6621a19bb0f559a85248955ac12c45a1a"
429
+ dependencies = [
430
+ "lock_api",
431
+ "parking_lot_core",
432
+ ]
433
+
434
+ [[package]]
435
+ name = "parking_lot_core"
436
+ version = "0.9.12"
437
+ source = "registry+https://github.com/rust-lang/crates.io-index"
438
+ checksum = "2621685985a2ebf1c516881c026032ac7deafcda1a2c9b7850dc81e3dfcb64c1"
439
+ dependencies = [
440
+ "cfg-if",
441
+ "libc",
442
+ "redox_syscall",
443
+ "smallvec",
444
+ "windows-link",
445
+ ]
446
+
447
+ [[package]]
448
+ name = "paste"
449
+ version = "1.0.15"
450
+ source = "registry+https://github.com/rust-lang/crates.io-index"
451
+ checksum = "57c0d7b74b563b49d38dae00a0c37d4d6de9b432382b2892f0574ddcae73fd0a"
452
+
453
+ [[package]]
454
+ name = "portable-atomic"
455
+ version = "1.13.1"
456
+ source = "registry+https://github.com/rust-lang/crates.io-index"
457
+ checksum = "c33a9471896f1c69cecef8d20cbe2f7accd12527ce60845ff44c153bb2a21b49"
458
+
459
+ [[package]]
460
+ name = "prettyplease"
461
+ version = "0.2.37"
462
+ source = "registry+https://github.com/rust-lang/crates.io-index"
463
+ checksum = "479ca8adacdd7ce8f1fb39ce9ecccbfe93a3f1344b3d0d97f20bc0196208f62b"
464
+ dependencies = [
465
+ "proc-macro2",
466
+ "syn 2.0.119",
467
+ ]
468
+
469
+ [[package]]
470
+ name = "proc-macro-crate"
471
+ version = "3.5.0"
472
+ source = "registry+https://github.com/rust-lang/crates.io-index"
473
+ checksum = "e67ba7e9b2b56446f1d419b1d807906278ffa1a658a8a5d8a39dcb1f5a78614f"
474
+ dependencies = [
475
+ "toml_edit",
476
+ ]
477
+
478
+ [[package]]
479
+ name = "proc-macro2"
480
+ version = "1.0.107"
481
+ source = "registry+https://github.com/rust-lang/crates.io-index"
482
+ checksum = "985e7ec9bb745e6ce6535b544d84d6cd6f7ad8bd711c398938ae983b91a766d9"
483
+ dependencies = [
484
+ "unicode-ident",
485
+ ]
486
+
487
+ [[package]]
488
+ name = "pyo3"
489
+ version = "0.29.2"
490
+ source = "registry+https://github.com/rust-lang/crates.io-index"
491
+ checksum = "4688ddedf473e32662b9b067670129a8afb8c18e351482c70d62ba4a88171e8b"
492
+ dependencies = [
493
+ "libc",
494
+ "once_cell",
495
+ "portable-atomic",
496
+ "pyo3-build-config",
497
+ "pyo3-ffi",
498
+ "pyo3-macros",
499
+ ]
500
+
501
+ [[package]]
502
+ name = "pyo3-build-config"
503
+ version = "0.29.2"
504
+ source = "registry+https://github.com/rust-lang/crates.io-index"
505
+ checksum = "f41027e41b4bd03f6e60f9f417fe24a6341a6bb744edd62b6f709f2a52ea30e9"
506
+ dependencies = [
507
+ "target-lexicon",
508
+ ]
509
+
510
+ [[package]]
511
+ name = "pyo3-ffi"
512
+ version = "0.29.2"
513
+ source = "registry+https://github.com/rust-lang/crates.io-index"
514
+ checksum = "e591a95526fead067432c3b3a33fc74770b87b1e04e73671090d9c2055a2b327"
515
+ dependencies = [
516
+ "libc",
517
+ "pyo3-build-config",
518
+ ]
519
+
520
+ [[package]]
521
+ name = "pyo3-macros"
522
+ version = "0.29.2"
523
+ source = "registry+https://github.com/rust-lang/crates.io-index"
524
+ checksum = "73225868fc1cd84eef2c3c230ddb91273bf1de46aeb8a4248da76d32a0924a1c"
525
+ dependencies = [
526
+ "proc-macro2",
527
+ "pyo3-macros-backend",
528
+ "quote",
529
+ "syn 2.0.119",
530
+ ]
531
+
532
+ [[package]]
533
+ name = "pyo3-macros-backend"
534
+ version = "0.29.2"
535
+ source = "registry+https://github.com/rust-lang/crates.io-index"
536
+ checksum = "571575aa3749fa6216757dd47d2a3e7ef360f329a40f0666a9fbd14889024952"
537
+ dependencies = [
538
+ "heck",
539
+ "proc-macro2",
540
+ "quote",
541
+ "syn 2.0.119",
542
+ ]
543
+
544
+ [[package]]
545
+ name = "quabla-core"
546
+ version = "0.6.1"
547
+ dependencies = [
548
+ "cudarc",
549
+ "libloading 0.9.0",
550
+ "libm",
551
+ "mlx-rs",
552
+ "mlx-sys",
553
+ "quabla-macros",
554
+ ]
555
+
556
+ [[package]]
557
+ name = "quabla-macros"
558
+ version = "0.6.1"
559
+ dependencies = [
560
+ "proc-macro2",
561
+ "quote",
562
+ "syn 3.0.6",
563
+ ]
564
+
565
+ [[package]]
566
+ name = "quabla-python"
567
+ version = "0.6.1"
568
+ dependencies = [
569
+ "pyo3",
570
+ "quabla-core",
571
+ ]
572
+
573
+ [[package]]
574
+ name = "quote"
575
+ version = "1.0.47"
576
+ source = "registry+https://github.com/rust-lang/crates.io-index"
577
+ checksum = "1fbf4db142a473a8d80c26bbf18454ed458bf8d26c8219c331daecfdbd079001"
578
+ dependencies = [
579
+ "proc-macro2",
580
+ ]
581
+
582
+ [[package]]
583
+ name = "redox_syscall"
584
+ version = "0.5.18"
585
+ source = "registry+https://github.com/rust-lang/crates.io-index"
586
+ checksum = "ed2bf2547551a7053d6fdfafda3f938979645c44812fbfcda098faae3f1a362d"
587
+ dependencies = [
588
+ "bitflags",
589
+ ]
590
+
591
+ [[package]]
592
+ name = "regex"
593
+ version = "1.13.0"
594
+ source = "registry+https://github.com/rust-lang/crates.io-index"
595
+ checksum = "2a0e75113e14dc5acb068cd0786884f214f1312650a3d36d269f5c4f3cdee8a2"
596
+ dependencies = [
597
+ "aho-corasick",
598
+ "memchr",
599
+ "regex-automata",
600
+ "regex-syntax",
601
+ ]
602
+
603
+ [[package]]
604
+ name = "regex-automata"
605
+ version = "0.4.15"
606
+ source = "registry+https://github.com/rust-lang/crates.io-index"
607
+ checksum = "1f388202e4b80542a0921078cc23b6333bcf1409c1e3f86404cae4766a6131db"
608
+ dependencies = [
609
+ "aho-corasick",
610
+ "memchr",
611
+ "regex-syntax",
612
+ ]
613
+
614
+ [[package]]
615
+ name = "regex-syntax"
616
+ version = "0.8.11"
617
+ source = "registry+https://github.com/rust-lang/crates.io-index"
618
+ checksum = "d6f6ff9a378485b298a5286656da665ba74413d36db0979633275d2e708145d4"
619
+
620
+ [[package]]
621
+ name = "rustc-hash"
622
+ version = "2.1.3"
623
+ source = "registry+https://github.com/rust-lang/crates.io-index"
624
+ checksum = "6b1e7f9a428571be2dc5bc0505c13fb6bf936822b894ec87abf8a08a4e51742d"
625
+
626
+ [[package]]
627
+ name = "rustversion"
628
+ version = "1.0.23"
629
+ source = "registry+https://github.com/rust-lang/crates.io-index"
630
+ checksum = "cf54715a573b99ac80df0bc206da022bcd442c974952c7b9720069370852e21f"
631
+
632
+ [[package]]
633
+ name = "scopeguard"
634
+ version = "1.2.0"
635
+ source = "registry+https://github.com/rust-lang/crates.io-index"
636
+ checksum = "94143f37725109f92c262ed2cf5e59bce7498c01bcc1502d7b9afe439a4e9f49"
637
+
638
+ [[package]]
639
+ name = "serde_core"
640
+ version = "1.0.228"
641
+ source = "registry+https://github.com/rust-lang/crates.io-index"
642
+ checksum = "41d385c7d4ca58e59fc732af25c3983b67ac852c1a25000afe1175de458b67ad"
643
+ dependencies = [
644
+ "serde_derive",
645
+ ]
646
+
647
+ [[package]]
648
+ name = "serde_derive"
649
+ version = "1.0.228"
650
+ source = "registry+https://github.com/rust-lang/crates.io-index"
651
+ checksum = "d540f220d3187173da220f885ab66608367b6574e925011a9353e4badda91d79"
652
+ dependencies = [
653
+ "proc-macro2",
654
+ "quote",
655
+ "syn 2.0.119",
656
+ ]
657
+
658
+ [[package]]
659
+ name = "shlex"
660
+ version = "1.3.0"
661
+ source = "registry+https://github.com/rust-lang/crates.io-index"
662
+ checksum = "0fda2ff0d084019ba4d7c6f371c95d8fd75ce3524c3cb8fb653a3023f6323e64"
663
+
664
+ [[package]]
665
+ name = "shlex"
666
+ version = "2.0.1"
667
+ source = "registry+https://github.com/rust-lang/crates.io-index"
668
+ checksum = "f8fadd59c855ef2080decdef8ff161eb6661b86933c9d82e5ba29dc602a55aba"
669
+
670
+ [[package]]
671
+ name = "smallvec"
672
+ version = "1.15.2"
673
+ source = "registry+https://github.com/rust-lang/crates.io-index"
674
+ checksum = "8ed6a63f02c8539c91a8685a86f4099661ba3da017932f6ebbea6de3f0fa7c90"
675
+
676
+ [[package]]
677
+ name = "strsim"
678
+ version = "0.11.1"
679
+ source = "registry+https://github.com/rust-lang/crates.io-index"
680
+ checksum = "7da8b5736845d9f2fcb837ea5d9e2628564b3b043a70948a3f0b778838c5fb4f"
681
+
682
+ [[package]]
683
+ name = "strum"
684
+ version = "0.28.0"
685
+ source = "registry+https://github.com/rust-lang/crates.io-index"
686
+ checksum = "9628de9b8791db39ceda2b119bbe13134770b56c138ec1d3af810d045c04f9bd"
687
+ dependencies = [
688
+ "strum_macros",
689
+ ]
690
+
691
+ [[package]]
692
+ name = "strum_macros"
693
+ version = "0.28.0"
694
+ source = "registry+https://github.com/rust-lang/crates.io-index"
695
+ checksum = "ab85eea0270ee17587ed4156089e10b9e6880ee688791d45a905f5b1ca36f664"
696
+ dependencies = [
697
+ "heck",
698
+ "proc-macro2",
699
+ "quote",
700
+ "syn 2.0.119",
701
+ ]
702
+
703
+ [[package]]
704
+ name = "syn"
705
+ version = "2.0.119"
706
+ source = "registry+https://github.com/rust-lang/crates.io-index"
707
+ checksum = "872831b642d1a07999a962a351ed35b955ea2cfc8f3862091e2a240a84f17297"
708
+ dependencies = [
709
+ "proc-macro2",
710
+ "quote",
711
+ "unicode-ident",
712
+ ]
713
+
714
+ [[package]]
715
+ name = "syn"
716
+ version = "3.0.6"
717
+ source = "registry+https://github.com/rust-lang/crates.io-index"
718
+ checksum = "8593e8e72159ed2257d083c7a454a85cbf854f37a0966d8d483aff8c8a3ebcee"
719
+ dependencies = [
720
+ "proc-macro2",
721
+ "quote",
722
+ "unicode-ident",
723
+ ]
724
+
725
+ [[package]]
726
+ name = "target-lexicon"
727
+ version = "0.13.5"
728
+ source = "registry+https://github.com/rust-lang/crates.io-index"
729
+ checksum = "adb6935a6f5c20170eeceb1a3835a49e12e19d792f6dd344ccc76a985ca5a6ca"
730
+
731
+ [[package]]
732
+ name = "thiserror"
733
+ version = "2.0.18"
734
+ source = "registry+https://github.com/rust-lang/crates.io-index"
735
+ checksum = "4288b5bcbc7920c07a1149a35cf9590a2aa808e0bc1eafaade0b80947865fbc4"
736
+ dependencies = [
737
+ "thiserror-impl",
738
+ ]
739
+
740
+ [[package]]
741
+ name = "thiserror-impl"
742
+ version = "2.0.18"
743
+ source = "registry+https://github.com/rust-lang/crates.io-index"
744
+ checksum = "ebc4ee7f67670e9b64d05fa4253e753e016c6c95ff35b89b7941d6b856dec1d5"
745
+ dependencies = [
746
+ "proc-macro2",
747
+ "quote",
748
+ "syn 2.0.119",
749
+ ]
750
+
751
+ [[package]]
752
+ name = "toml_datetime"
753
+ version = "1.1.1+spec-1.1.0"
754
+ source = "registry+https://github.com/rust-lang/crates.io-index"
755
+ checksum = "3165f65f62e28e0115a00b2ebdd37eb6f3b641855f9d636d3cd4103767159ad7"
756
+ dependencies = [
757
+ "serde_core",
758
+ ]
759
+
760
+ [[package]]
761
+ name = "toml_edit"
762
+ version = "0.25.12+spec-1.1.0"
763
+ source = "registry+https://github.com/rust-lang/crates.io-index"
764
+ checksum = "d2153edc6955a6c354fad8f5efd38b6a8769bdccf9fe50f8e1329f81b0baa5d7"
765
+ dependencies = [
766
+ "indexmap",
767
+ "toml_datetime",
768
+ "toml_parser",
769
+ "winnow",
770
+ ]
771
+
772
+ [[package]]
773
+ name = "toml_parser"
774
+ version = "1.1.2+spec-1.1.0"
775
+ source = "registry+https://github.com/rust-lang/crates.io-index"
776
+ checksum = "a2abe9b86193656635d2411dc43050282ca48aa31c2451210f4202550afb7526"
777
+ dependencies = [
778
+ "winnow",
779
+ ]
780
+
781
+ [[package]]
782
+ name = "unicode-ident"
783
+ version = "1.0.24"
784
+ source = "registry+https://github.com/rust-lang/crates.io-index"
785
+ checksum = "e6e4313cd5fcd3dad5cafa179702e2b244f760991f45397d14d4ebf38247da75"
786
+
787
+ [[package]]
788
+ name = "windows-link"
789
+ version = "0.2.1"
790
+ source = "registry+https://github.com/rust-lang/crates.io-index"
791
+ checksum = "f0805222e57f7521d6a62e36fa9163bc891acd422f971defe97d64e70d0a4fe5"
792
+
793
+ [[package]]
794
+ name = "winnow"
795
+ version = "1.0.3"
796
+ source = "registry+https://github.com/rust-lang/crates.io-index"
797
+ checksum = "0592e1c9d151f854e6fd382574c3a0855250e1d9b2f99d9281c6e6391af352f1"
798
+ dependencies = [
799
+ "memchr",
800
+ ]
801
+
802
+ [[package]]
803
+ name = "zerocopy"
804
+ version = "0.8.54"
805
+ source = "registry+https://github.com/rust-lang/crates.io-index"
806
+ checksum = "b7cbbc0a705a0fd05cc3676525980d2bf5a9bc4adac6d6475209a7887cf59d19"
807
+ dependencies = [
808
+ "zerocopy-derive",
809
+ ]
810
+
811
+ [[package]]
812
+ name = "zerocopy-derive"
813
+ version = "0.8.54"
814
+ source = "registry+https://github.com/rust-lang/crates.io-index"
815
+ checksum = "e2e817b7b52d0c7358d3246da9d69935ebb18116b2b102b4230dac079b4862f5"
816
+ dependencies = [
817
+ "proc-macro2",
818
+ "quote",
819
+ "syn 2.0.119",
820
+ ]