mlx-bun 0.4.0 → 1.0.0

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 (601) hide show
  1. package/README.md +1453 -88
  2. package/THIRD_PARTY_NOTICES.md +882 -0
  3. package/bin/mlx-bun.convert.mjs +4 -0
  4. package/bin/mlx-bun.fuse.mjs +4 -0
  5. package/bin/mlx-bun.generate.mjs +4 -0
  6. package/bin/mlx-bun.lora.mjs +4 -0
  7. package/bin/mlx-bun.mjs +13 -21
  8. package/bin/mlx-bun.server.mjs +4 -0
  9. package/bin/mlx-bun.upload.mjs +4 -0
  10. package/dist/web/app.js +9584 -0
  11. package/package.json +50 -53
  12. package/src/cli/ane-install.ts +76 -0
  13. package/src/cli/args.ts +179 -0
  14. package/src/cli/engine-entry.ts +68 -0
  15. package/src/cli/inference.ts +214 -0
  16. package/src/cli/job-entry.ts +97 -0
  17. package/src/cli/library-host.ts +117 -0
  18. package/src/cli/main.ts +57 -0
  19. package/src/cli/memory-engine.ts +163 -0
  20. package/src/cli/memory.ts +36 -0
  21. package/src/cli/mlx-lm-aliases.ts +309 -0
  22. package/src/cli/model-choice.ts +33 -0
  23. package/src/cli/model-selection.ts +84 -0
  24. package/src/cli/module-host.ts +11 -0
  25. package/src/cli/module-verbs.ts +50 -0
  26. package/src/cli/numerical-policy.ts +12 -0
  27. package/src/cli/perplexity.ts +89 -0
  28. package/src/cli/serve-host.ts +423 -0
  29. package/src/cli/serve-isolated.ts +217 -0
  30. package/src/cli/serve-options.ts +106 -0
  31. package/src/cli/serve-state.ts +205 -0
  32. package/src/cli/serve.ts +500 -0
  33. package/src/cli/served-model-host.ts +102 -0
  34. package/src/cli/server-entry.ts +148 -0
  35. package/src/cli/serving-unit.ts +177 -0
  36. package/src/{tui.ts → cli/terminal.ts} +13 -8
  37. package/src/cli/worker-adapters.ts +34 -0
  38. package/src/cli/worker-entry.ts +272 -0
  39. package/src/cli/worker-unit.ts +179 -0
  40. package/src/engine/adapter-operation.ts +26 -0
  41. package/src/engine/cache-services.ts +114 -0
  42. package/src/engine/completion.ts +39 -0
  43. package/src/engine/generation-gateway.ts +358 -0
  44. package/src/engine/index.ts +59 -0
  45. package/src/engine/kv-budget.ts +120 -0
  46. package/src/engine/media-preparation.ts +232 -0
  47. package/src/engine/model-binding.ts +24 -0
  48. package/src/engine/model-host.ts +513 -0
  49. package/src/engine/model-serving.ts +89 -0
  50. package/src/{serve → engine}/preparation.ts +10 -8
  51. package/src/{serve → engine}/session-completion-engine.ts +6 -6
  52. package/src/engine/telemetry.ts +109 -0
  53. package/src/hub/downloads.ts +116 -0
  54. package/src/jobs/db.ts +16 -7
  55. package/src/jobs/events.ts +29 -0
  56. package/src/jobs/executable.ts +5 -0
  57. package/src/jobs/host.ts +51 -0
  58. package/src/jobs/in-process.ts +55 -0
  59. package/src/{contracts/jobs.ts → jobs/protocol.ts} +7 -9
  60. package/src/jobs/runner.ts +150 -231
  61. package/src/jobs/service.ts +75 -0
  62. package/src/jobs/sse.ts +59 -23
  63. package/src/jobs/worker-memory.ts +46 -0
  64. package/src/jobs/worker-process.ts +163 -0
  65. package/src/jobs/worker-supervisor.ts +258 -0
  66. package/src/modules.ts +58 -0
  67. package/src/publishing/catalog-hub.ts +18 -0
  68. package/src/publishing/credentials.ts +31 -0
  69. package/src/publishing/upload.ts +37 -0
  70. package/src/residency/leased-adapters.ts +21 -0
  71. package/src/residency/local-records.ts +16 -0
  72. package/src/residency/model-residency.ts +405 -0
  73. package/src/residency/record-index.ts +28 -0
  74. package/src/residency/resident-estimate.ts +50 -0
  75. package/src/{anthropic.ts → server/anthropic.ts} +3 -81
  76. package/src/server/cache-routes.ts +36 -0
  77. package/src/{serve → server}/chat-request.ts +12 -10
  78. package/src/{serve → server}/chat-stage.ts +27 -22
  79. package/src/{client.ts → server/client.ts} +28 -2
  80. package/src/{serve → server}/completion-executor.ts +42 -42
  81. package/src/server/decision-body.ts +55 -0
  82. package/src/server/decision-routes.ts +134 -0
  83. package/src/server/decision-wire.ts +228 -0
  84. package/src/{serve → server}/discovery-routes.ts +42 -83
  85. package/src/server/generated-token-history.ts +57 -0
  86. package/src/{serve → server}/http.ts +25 -13
  87. package/src/{serve → server}/inference-request.ts +5 -5
  88. package/src/server/job-routes.ts +20 -0
  89. package/src/server/media-prompt.ts +63 -0
  90. package/src/server/memory-completion-client.ts +46 -0
  91. package/src/server/model-routes.ts +192 -0
  92. package/src/{serve → server}/openai-wire.ts +3 -2
  93. package/src/{serve → server}/pipeline.ts +1 -1
  94. package/src/server/prompt-builder.ts +13 -0
  95. package/src/server/prompt-contracts.ts +17 -0
  96. package/src/server/proxy-routes.ts +187 -0
  97. package/src/server/publishing-routes.ts +41 -0
  98. package/src/{serve → server}/request-plan.ts +8 -8
  99. package/src/{serve → server}/request-prep.ts +82 -68
  100. package/src/{serve → server}/responses-client.ts +30 -13
  101. package/src/{responses.ts → server/responses.ts} +12 -93
  102. package/src/server/routes.ts +205 -0
  103. package/src/server/start.ts +72 -0
  104. package/src/server/status-routes.ts +108 -0
  105. package/src/{serve → server}/text-completion-stage.ts +25 -4
  106. package/src/server/text-prompt.ts +12 -0
  107. package/src/{serve → server}/token-streams.ts +36 -38
  108. package/src/server/worker-routes.ts +311 -0
  109. package/src/storage/paths.ts +61 -0
  110. package/src/web/assets.ts +37 -0
  111. package/src/web/{src → browser}/api.ts +5 -23
  112. package/src/web/browser/chat-host.ts +17 -0
  113. package/src/web/browser/installed-panels.ts +6 -0
  114. package/src/web/browser/main.ts +55 -0
  115. package/src/web/browser/markdown.ts +279 -0
  116. package/src/web/browser/memory-panel.ts +9 -0
  117. package/src/web/browser/palette.ts +64 -0
  118. package/src/web/browser/protocol.ts +13 -0
  119. package/src/web/browser/shell.ts +467 -0
  120. package/src/web/{src → browser}/status.ts +32 -36
  121. package/src/web/build.ts +51 -0
  122. package/src/web/public/app.html +1054 -0
  123. package/src/web/{sw.js → public/sw.js} +4 -3
  124. package/src/web/{vendor → public/vendor}/hljs-theme.css +2 -2
  125. package/src/web/{vendor → public/vendor}/hljs.js +2 -2
  126. package/THIRD_PARTY_LICENSES.md +0 -51
  127. package/docs/README.md +0 -94
  128. package/docs/archive/hlg-sampling.md +0 -350
  129. package/docs/archive/investigations/audio-input-plan.md +0 -356
  130. package/docs/archive/investigations/chunk-finetune-experiment.md +0 -84
  131. package/docs/archive/investigations/colibri-glm52-port.md +0 -1486
  132. package/docs/archive/investigations/colibri-oracle-pin.md +0 -1284
  133. package/docs/archive/investigations/curve-bisector-routing.md +0 -123
  134. package/docs/archive/investigations/decode-roofline-lookagain.md +0 -348
  135. package/docs/archive/investigations/diffusion-gemma-port.md +0 -602
  136. package/docs/archive/investigations/dspark-handoff.md +0 -224
  137. package/docs/archive/investigations/expert-offload-single-user-moe.md +0 -292
  138. package/docs/archive/investigations/faithful-l1-consolidation.md +0 -310
  139. package/docs/archive/investigations/hlg-sampling-investigation.md +0 -970
  140. package/docs/archive/investigations/indexshare-performance-spike.md +0 -357
  141. package/docs/archive/investigations/kernel-perf-review-2026-07.md +0 -171
  142. package/docs/archive/investigations/lab-build-journal.md +0 -157
  143. package/docs/archive/investigations/minicpm5-decode-megakernel.md +0 -489
  144. package/docs/archive/investigations/openjarvis-fit.md +0 -198
  145. package/docs/archive/investigations/openwiki-evaluation.md +0 -249
  146. package/docs/archive/investigations/optimization_plan.md +0 -346
  147. package/docs/archive/investigations/orpo-base-uf-experiment-and-directions.md +0 -120
  148. package/docs/archive/investigations/orpo-flash-cce-pin-leak.md +0 -167
  149. package/docs/archive/investigations/orpo-uf-testing-handoff.md +0 -126
  150. package/docs/archive/investigations/pareto-specialized-runtime-findings.md +0 -92
  151. package/docs/archive/investigations/pi-builtin-investigation.md +0 -308
  152. package/docs/archive/investigations/pre-colibri-stabilization.md +0 -329
  153. package/docs/archive/investigations/qwen38-27b-campaign-log.md +0 -1092
  154. package/docs/archive/investigations/segmented-backward-handoff.md +0 -148
  155. package/docs/archive/investigations/starter-model-port-handoff.md +0 -342
  156. package/docs/archive/investigations/steel-flash-cce-handoff.md +0 -489
  157. package/docs/archive/investigations/the-dreaming-handoff.md +0 -141
  158. package/docs/archive/investigations/trainer-validation-experiment.md +0 -173
  159. package/docs/archive/planning/chat-ui-vision-memory-first.md +0 -199
  160. package/docs/archive/planning/chat-ui-vision.md +0 -50
  161. package/docs/archive/planning/chunk-task-roadmap.md +0 -77
  162. package/docs/archive/planning/curve-sampler-research-plan.md +0 -331
  163. package/docs/archive/planning/journal.md +0 -316
  164. package/docs/archive/planning/memory-docs-and-dag-plan.md +0 -194
  165. package/docs/archive/planning/pi-integration-review.md +0 -178
  166. package/docs/archive/planning/release-notes-v0.0.10.md +0 -57
  167. package/docs/archive/planning/release-notes-v0.0.11.md +0 -239
  168. package/docs/archive/planning/release-notes-v0.0.12.md +0 -80
  169. package/docs/archive/planning/release-notes-v0.0.13.md +0 -77
  170. package/docs/archive/planning/release-notes-v0.0.9.md +0 -122
  171. package/docs/archive/planning/release-notes-v0.2.0.md +0 -49
  172. package/docs/archive/planning/release-notes-v0.3.0.md +0 -35
  173. package/docs/archive/planning/web-ui-pass-plan.md +0 -213
  174. package/docs/archive/planning/website-readme-pass-plan.md +0 -224
  175. package/docs/design/batching.md +0 -1841
  176. package/docs/design/decode-speed-program.md +0 -2349
  177. package/docs/design/dreaming-nightly-pipeline.md +0 -684
  178. package/docs/design/generic-model-support.md +0 -705
  179. package/docs/design/kv-cache.md +0 -824
  180. package/docs/design/orpo-training.md +0 -680
  181. package/docs/design/speculative-decoding.md +0 -1990
  182. package/docs/design/turboquant.md +0 -2212
  183. package/docs/design/unified-engine-frontier-plan.md +0 -1880
  184. package/docs/design/web-chat-redesign.md +0 -695
  185. package/docs/planning/PRODUCT_ROADMAP.md +0 -594
  186. package/docs/planning/ResearchTopics.md +0 -119
  187. package/docs/planning/release-notes-v0.4.0.md +0 -45
  188. package/docs/planning/video-series.md +0 -421
  189. package/docs/reference/benchmarks.md +0 -3613
  190. package/docs/reference/cli.md +0 -585
  191. package/docs/reference/distribution.md +0 -355
  192. package/docs/reference/environment.md +0 -398
  193. package/docs/reference/glossary.md +0 -39
  194. package/docs/reference/library-api.md +0 -674
  195. package/docs/reference/memory.md +0 -260
  196. package/docs/reference/models.md +0 -334
  197. package/docs/reference/server-api.md +0 -1775
  198. package/docs/reference/server-config.md +0 -846
  199. package/docs/reference/training.md +0 -462
  200. package/docs/reference/troubleshooting.md +0 -158
  201. package/src/admission-plan.ts +0 -18
  202. package/src/audio/conformer.ts +0 -647
  203. package/src/audio/decode.ts +0 -133
  204. package/src/audio/features.ts +0 -202
  205. package/src/audio/transcode.ts +0 -67
  206. package/src/backends/mlx/assistant-target.ts +0 -46
  207. package/src/backends/mlx/autoregressive-draft-rows.ts +0 -120
  208. package/src/backends/mlx/autoregressive.ts +0 -108
  209. package/src/backends/mlx/batch-group.ts +0 -1790
  210. package/src/backends/mlx/cache-attachment.ts +0 -67
  211. package/src/backends/mlx/cache-layout.ts +0 -31
  212. package/src/backends/mlx/checkpoint-state.ts +0 -53
  213. package/src/backends/mlx/continuation-persistence.ts +0 -106
  214. package/src/backends/mlx/continuation-request.ts +0 -43
  215. package/src/backends/mlx/continuation.ts +0 -77
  216. package/src/backends/mlx/diffusion.ts +0 -39
  217. package/src/backends/mlx/draft-target.ts +0 -32
  218. package/src/backends/mlx/evaluation.ts +0 -105
  219. package/src/backends/mlx/fill-append.ts +0 -45
  220. package/src/backends/mlx/full-kv-row-append.ts +0 -32
  221. package/src/backends/mlx/full-kv-row-donor.ts +0 -54
  222. package/src/backends/mlx/full-transitioning-kv-rows.ts +0 -65
  223. package/src/backends/mlx/gateway-binding.ts +0 -130
  224. package/src/backends/mlx/graph.ts +0 -38
  225. package/src/backends/mlx/kv-codec.ts +0 -9
  226. package/src/backends/mlx/kv-maintenance.ts +0 -133
  227. package/src/backends/mlx/kv-tensor-rows.ts +0 -212
  228. package/src/backends/mlx/memory-client.ts +0 -189
  229. package/src/backends/mlx/methods.ts +0 -90
  230. package/src/backends/mlx/model-prompt.ts +0 -164
  231. package/src/backends/mlx/model-serving.ts +0 -79
  232. package/src/backends/mlx/prefill-cohort.ts +0 -89
  233. package/src/backends/mlx/prefill-rows.ts +0 -200
  234. package/src/backends/mlx/prefill.ts +0 -33
  235. package/src/backends/mlx/projected-context-rows.ts +0 -72
  236. package/src/backends/mlx/request-policy.ts +0 -20
  237. package/src/backends/mlx/request-state-policy.ts +0 -41
  238. package/src/backends/mlx/rollback.ts +0 -89
  239. package/src/backends/mlx/row-sampling.ts +0 -49
  240. package/src/backends/mlx/serial-executor.ts +0 -242
  241. package/src/backends/mlx/speculative-group.ts +0 -359
  242. package/src/backends/mlx/speculative-round.ts +0 -161
  243. package/src/backends/mlx/speculative.ts +0 -114
  244. package/src/backends/mlx/state-rows.ts +0 -104
  245. package/src/backends/mlx/state-views.ts +0 -46
  246. package/src/backends/mlx/target-layout-capability.ts +0 -35
  247. package/src/backends/mlx/text-engine.ts +0 -73
  248. package/src/backends/mlx/transitioning-kv-rows.ts +0 -165
  249. package/src/chat-template.ts +0 -382
  250. package/src/cli.ts +0 -3168
  251. package/src/config.ts +0 -399
  252. package/src/contracts/completion.ts +0 -16
  253. package/src/contracts/execution.ts +0 -60
  254. package/src/contracts/generation.ts +0 -103
  255. package/src/contracts/host.ts +0 -7
  256. package/src/contracts/pi-provider.ts +0 -30
  257. package/src/contracts/pi.ts +0 -259
  258. package/src/contracts/prefix-cache.ts +0 -23
  259. package/src/contracts/resources.ts +0 -25
  260. package/src/contracts/sampling.ts +0 -23
  261. package/src/contracts/scheduling.ts +0 -45
  262. package/src/contracts/tsconfig.json +0 -14
  263. package/src/dataset/generators.ts +0 -1087
  264. package/src/dataset/index.ts +0 -12
  265. package/src/dataset/job.ts +0 -81
  266. package/src/dataset/llm.ts +0 -71
  267. package/src/dataset/registry.ts +0 -528
  268. package/src/diffusion/diffusion-generate.ts +0 -507
  269. package/src/download.ts +0 -505
  270. package/src/embed.ts +0 -77
  271. package/src/engine/admission.ts +0 -59
  272. package/src/engine/cancellation.ts +0 -38
  273. package/src/engine/engine.ts +0 -56
  274. package/src/engine/execution-coordinator.ts +0 -70
  275. package/src/engine/execution-plan.ts +0 -53
  276. package/src/engine/resources.ts +0 -53
  277. package/src/engine/scheduler.ts +0 -72
  278. package/src/engine/session.ts +0 -247
  279. package/src/eval/README.md +0 -130
  280. package/src/eval/ifeval.ts +0 -81
  281. package/src/eval/kl-prompts.ts +0 -29
  282. package/src/eval/kl.ts +0 -590
  283. package/src/eval/perplexity.ts +0 -164
  284. package/src/eval/quality-db.ts +0 -92
  285. package/src/eval/reference-logits.ts +0 -207
  286. package/src/eval/runner.ts +0 -124
  287. package/src/eval/score.ts +0 -49
  288. package/src/eval/tasks/bfcl.ts +0 -535
  289. package/src/eval/tasks/gsm8k.ts +0 -98
  290. package/src/eval/tasks/hashhop.ts +0 -139
  291. package/src/eval/tasks/humaneval.ts +0 -275
  292. package/src/eval/tasks/ifeval.ts +0 -525
  293. package/src/eval/tasks/mmlu.ts +0 -157
  294. package/src/eval/ultrafeedback.ts +0 -60
  295. package/src/evaldb.ts +0 -199
  296. package/src/expert-io.ts +0 -393
  297. package/src/expert-offload-build.ts +0 -77
  298. package/src/expert-offload.ts +0 -76
  299. package/src/expert-residency.ts +0 -1134
  300. package/src/expert-usage.ts +0 -415
  301. package/src/fill/echo-index.ts +0 -188
  302. package/src/fill/fill-session.ts +0 -452
  303. package/src/fill/proposal.ts +0 -68
  304. package/src/fill/schema-rows.ts +0 -506
  305. package/src/fill/tool-boundary.ts +0 -103
  306. package/src/fit.ts +0 -249
  307. package/src/generate.ts +0 -1346
  308. package/src/grammar.ts +0 -591
  309. package/src/harness-pi.ts +0 -184
  310. package/src/hf-push.ts +0 -628
  311. package/src/hub-rest.ts +0 -242
  312. package/src/index.ts +0 -63
  313. package/src/inference/denoising.ts +0 -17
  314. package/src/inference/draft-acceptance.ts +0 -41
  315. package/src/inference/draft-output.ts +0 -49
  316. package/src/inference/graph.ts +0 -33
  317. package/src/inference/prefill.ts +0 -39
  318. package/src/inference/rollback.ts +0 -7
  319. package/src/jobs/index.ts +0 -31
  320. package/src/jobs/job-entry.ts +0 -107
  321. package/src/jobs/task-client.ts +0 -19
  322. package/src/jobs/types.ts +0 -2
  323. package/src/kv-scheme.ts +0 -257
  324. package/src/kv-store.ts +0 -1099
  325. package/src/lab/README.md +0 -18
  326. package/src/lab/curve/curve-designer.html +0 -303
  327. package/src/lab/curve/curve-sampler.ts +0 -122
  328. package/src/lab/expert-trace/expert-trace.ts +0 -69
  329. package/src/lab/paged-kv/paged-kv-rows.ts +0 -83
  330. package/src/lab/paged-kv/paged-kv.ts +0 -270
  331. package/src/library.ts +0 -40
  332. package/src/lora.ts +0 -408
  333. package/src/md-modules.d.ts +0 -49
  334. package/src/media-fetch.ts +0 -442
  335. package/src/memory/article.ts +0 -558
  336. package/src/memory/chunk-validation.ts +0 -150
  337. package/src/memory/chunk.ts +0 -318
  338. package/src/memory/cluster.ts +0 -351
  339. package/src/memory/coverage.ts +0 -134
  340. package/src/memory/crosslink.ts +0 -0
  341. package/src/memory/db.ts +0 -253
  342. package/src/memory/entity.ts +0 -203
  343. package/src/memory/gate.ts +0 -187
  344. package/src/memory/history.ts +0 -152
  345. package/src/memory/ingest.ts +0 -335
  346. package/src/memory/model.ts +0 -183
  347. package/src/memory/normalize.ts +0 -671
  348. package/src/memory/parse.ts +0 -22
  349. package/src/memory/pipeline.ts +0 -275
  350. package/src/memory/prompts.ts +0 -65
  351. package/src/memory/query.ts +0 -334
  352. package/src/memory/reconcile.ts +0 -734
  353. package/src/memory/redirect.ts +0 -108
  354. package/src/memory/reindex.ts +0 -325
  355. package/src/memory/resolve.ts +0 -307
  356. package/src/memory/rest.ts +0 -245
  357. package/src/memory/route.ts +0 -272
  358. package/src/memory/schedule.ts +0 -165
  359. package/src/memory/stages.ts +0 -777
  360. package/src/memory/synthesize.ts +0 -1638
  361. package/src/memory/tools.ts +0 -749
  362. package/src/memory/vault.ts +0 -1194
  363. package/src/memory/wikify.ts +0 -1064
  364. package/src/mlx/array.ts +0 -388
  365. package/src/mlx/autograd.ts +0 -304
  366. package/src/mlx/checkpoint.ts +0 -149
  367. package/src/mlx/compile.ts +0 -168
  368. package/src/mlx/custom-vjp.ts +0 -188
  369. package/src/mlx/ffi.ts +0 -512
  370. package/src/mlx/materialize.ts +0 -23
  371. package/src/mlx/metal-kernel.ts +0 -159
  372. package/src/mlx/ops.ts +0 -1021
  373. package/src/mlx/quantized-rows.ts +0 -36
  374. package/src/mlx/token-bitmask.ts +0 -52
  375. package/src/mlx/turboquant-kv-decode.ts +0 -82
  376. package/src/mlx/turboquant-kv-inverse.ts +0 -122
  377. package/src/mlx/turboquant-ops.ts +0 -861
  378. package/src/mlx/turboquant-tables.ts +0 -237
  379. package/src/mmap.ts +0 -90
  380. package/src/model/artifact-identity.ts +0 -73
  381. package/src/model/batched-kv.ts +0 -235
  382. package/src/model/batched-mask.ts +0 -279
  383. package/src/model/batched-quant.ts +0 -212
  384. package/src/model/batched-quantized-kv.ts +0 -144
  385. package/src/model/batched-rotating-quant.ts +0 -353
  386. package/src/model/batched-rotating-state.ts +0 -149
  387. package/src/model/batched-rotating.ts +0 -338
  388. package/src/model/batched-row-storage.ts +0 -179
  389. package/src/model/batched-ssm.ts +0 -52
  390. package/src/model/batched-turboquant-kv.ts +0 -93
  391. package/src/model/compiled-decode.ts +0 -712
  392. package/src/model/decoded-kv-donor.ts +0 -12
  393. package/src/model/delayed-quantized-kv.ts +0 -72
  394. package/src/model/delayed-rotating-quantized-kv.ts +0 -93
  395. package/src/model/delayed-turboquant-kv.ts +0 -55
  396. package/src/model/diffusion-gemma.ts +0 -1026
  397. package/src/model/factory.ts +0 -189
  398. package/src/model/fingerprint.ts +0 -33
  399. package/src/model/flash-attention.ts +0 -694
  400. package/src/model/full-prefill-padding.ts +0 -56
  401. package/src/model/full-prefill-row.ts +0 -105
  402. package/src/model/gemma4-base.ts +0 -2351
  403. package/src/model/gemma4.ts +0 -1315
  404. package/src/model/generated/gemma4-12b.ts +0 -429
  405. package/src/model/generated/gemma4-26b.ts +0 -448
  406. package/src/model/generated/gemma4-e4b.ts +0 -717
  407. package/src/model/generated/index.ts +0 -18
  408. package/src/model/glm52-atlas.ts +0 -511
  409. package/src/model/glm52-cache.ts +0 -792
  410. package/src/model/glm52-config.ts +0 -262
  411. package/src/model/glm52-container.ts +0 -305
  412. package/src/model/glm52-coupling.ts +0 -388
  413. package/src/model/glm52-dsa.ts +0 -606
  414. package/src/model/glm52-expert-layout.ts +0 -253
  415. package/src/model/glm52-layout.ts +0 -152
  416. package/src/model/glm52-memory.ts +0 -500
  417. package/src/model/glm52-mla.ts +0 -792
  418. package/src/model/glm52-moe.ts +0 -223
  419. package/src/model/glm52-pilot.ts +0 -537
  420. package/src/model/glm52-reference.ts +0 -445
  421. package/src/model/glm52-residency.ts +0 -557
  422. package/src/model/glm52-resident-weights.ts +0 -135
  423. package/src/model/glm52-streamed-experts.ts +0 -453
  424. package/src/model/glm52-streamed-metal.ts +0 -610
  425. package/src/model/glm52-weights.ts +0 -503
  426. package/src/model/glm52.ts +0 -1132
  427. package/src/model/implementation.ts +0 -69
  428. package/src/model/kv-attention-view.ts +0 -67
  429. package/src/model/minicpm5.ts +0 -376
  430. package/src/model/profile.ts +0 -443
  431. package/src/model/qwen-append.ts +0 -12
  432. package/src/model/qwen-conv.ts +0 -52
  433. package/src/model/qwen3-delta-state.ts +0 -74
  434. package/src/model/qwen3-delta.ts +0 -451
  435. package/src/model/qwen3-moe.ts +0 -404
  436. package/src/model/qwen3-mrope.ts +0 -252
  437. package/src/model/qwen3.ts +0 -277
  438. package/src/model/qwen3_5-checkpoint.ts +0 -164
  439. package/src/model/qwen3_5.ts +0 -819
  440. package/src/model/rotating-kv-donor.ts +0 -35
  441. package/src/model/rotating-kv-layout.ts +0 -262
  442. package/src/model/rotating-row-transaction.ts +0 -35
  443. package/src/model/speculative-rotating-kv.ts +0 -101
  444. package/src/model/ssm-prefill-padding.ts +0 -39
  445. package/src/model/support.ts +0 -102
  446. package/src/model/trellis-balanced-scatter.ts +0 -51
  447. package/src/model/trellis-linear.ts +0 -672
  448. package/src/model/trellis-shared-m.ts +0 -151
  449. package/src/model/trellis-shared-scatter.ts +0 -144
  450. package/src/model/trellis-splitk-prefill.ts +0 -158
  451. package/src/model/trellis-tiled-prefill.ts +0 -136
  452. package/src/model/trellis-vector-expand.ts +0 -62
  453. package/src/model/trellis-wide-prefill.ts +0 -94
  454. package/src/model/turboquant-codec.ts +0 -83
  455. package/src/model/universal/archs.ts +0 -332
  456. package/src/model/universal/dense.ts +0 -612
  457. package/src/model/universal/modules.ts +0 -170
  458. package/src/model/universal/rope.ts +0 -381
  459. package/src/native/expert_io.c +0 -523
  460. package/src/native/frame_extract.swift +0 -140
  461. package/src/native-pack.ts +0 -162
  462. package/src/pi-provider.ts +0 -53
  463. package/src/pi-session.ts +0 -81
  464. package/src/pi-terminal.ts +0 -274
  465. package/src/pi-web.ts +0 -1932
  466. package/src/preflight.ts +0 -158
  467. package/src/prompt-cache.ts +0 -460
  468. package/src/quantize/allocator.ts +0 -627
  469. package/src/quantize/atomic-output.ts +0 -38
  470. package/src/quantize/calibration.ts +0 -181
  471. package/src/quantize/config-writer.ts +0 -145
  472. package/src/quantize/index.ts +0 -84
  473. package/src/quantize/job.ts +0 -162
  474. package/src/quantize/quantizer.ts +0 -666
  475. package/src/quantize/rotate.ts +0 -603
  476. package/src/quantize/safetensors-writer.ts +0 -249
  477. package/src/quantize/sensitivity.ts +0 -216
  478. package/src/quantize/trellis.ts +0 -586
  479. package/src/quantize/weight-transform.ts +0 -292
  480. package/src/registry.ts +0 -602
  481. package/src/runtime-config.ts +0 -81
  482. package/src/safetensors.ts +0 -158
  483. package/src/sampler.ts +0 -1011
  484. package/src/serve/admin-routes.ts +0 -184
  485. package/src/serve/assets/curve-terrain.html +0 -105
  486. package/src/serve/assets/training-inference-map.html +0 -575
  487. package/src/serve/aux-routes.ts +0 -132
  488. package/src/serve/batch-scheduler.ts +0 -8
  489. package/src/serve/checkpoint-identity.ts +0 -64
  490. package/src/serve/generation-gateway.ts +0 -496
  491. package/src/serve/isolate.ts +0 -609
  492. package/src/serve/kv-budget.ts +0 -14
  493. package/src/serve/lab-routes.ts +0 -341
  494. package/src/serve/lane-registry.ts +0 -45
  495. package/src/serve/model-admin-routes.ts +0 -155
  496. package/src/serve/model-binding.ts +0 -43
  497. package/src/serve/model-host.ts +0 -511
  498. package/src/serve/parent-application.ts +0 -56
  499. package/src/serve/prompt-response-trace.ts +0 -147
  500. package/src/serve/response-lease.ts +0 -24
  501. package/src/serve/session-search.ts +0 -326
  502. package/src/serve/static-routes.ts +0 -101
  503. package/src/server.ts +0 -1314
  504. package/src/spec/assistant-rows.ts +0 -92
  505. package/src/spec/assistant-source.ts +0 -126
  506. package/src/spec/cache-identity.ts +0 -22
  507. package/src/spec/deepspec-rows.ts +0 -29
  508. package/src/spec/deepspec-source.ts +0 -145
  509. package/src/spec/dflash-source.ts +0 -149
  510. package/src/spec/drafter.ts +0 -304
  511. package/src/spec/dspark/ab-stats.ts +0 -136
  512. package/src/spec/dspark/calibration.ts +0 -112
  513. package/src/spec/dspark/data-dflash.ts +0 -156
  514. package/src/spec/dspark/data.ts +0 -201
  515. package/src/spec/dspark/deepspec-module.ts +0 -776
  516. package/src/spec/dspark/generate-dflash.ts +0 -245
  517. package/src/spec/dspark/generate.ts +0 -290
  518. package/src/spec/dspark/loader.ts +0 -23
  519. package/src/spec/dspark/loss.ts +0 -213
  520. package/src/spec/dspark/module-dflash.ts +0 -560
  521. package/src/spec/dspark/module.ts +0 -540
  522. package/src/spec/dspark/quantize-drafter.ts +0 -91
  523. package/src/spec/dspark/sample.ts +0 -112
  524. package/src/spec/generate.ts +0 -276
  525. package/src/spec/glm52-mtp-graph.ts +0 -29
  526. package/src/spec/glm52-mtp-rows.ts +0 -132
  527. package/src/spec/glm52-mtp-source.ts +0 -317
  528. package/src/spec/glm52-mtp-state.ts +0 -33
  529. package/src/spec/ngram-source.ts +0 -204
  530. package/src/spec/projected-draft-rows.ts +0 -73
  531. package/src/spec/qwen-mtp-module.ts +0 -171
  532. package/src/spec/qwen-mtp-rows.ts +0 -219
  533. package/src/spec/qwen-mtp-source.ts +0 -383
  534. package/src/spec/qwen-mtp-state.ts +0 -30
  535. package/src/spec/serve-loop.ts +0 -585
  536. package/src/spec/source.ts +0 -215
  537. package/src/spec/two-model-rows.ts +0 -61
  538. package/src/spec/two-model.ts +0 -113
  539. package/src/ssd-cache.ts +0 -372
  540. package/src/ssd-durability.ts +0 -211
  541. package/src/tokenizer.ts +0 -335
  542. package/src/tool-approvals.ts +0 -131
  543. package/src/tool-call.ts +0 -695
  544. package/src/train/dataset.ts +0 -484
  545. package/src/train/diffusion-lora.ts +0 -216
  546. package/src/train/export.ts +0 -32
  547. package/src/train/flash-cce.ts +0 -1234
  548. package/src/train/forward.ts +0 -266
  549. package/src/train/fuse.ts +0 -243
  550. package/src/train/index.ts +0 -46
  551. package/src/train/job.ts +0 -205
  552. package/src/train/lora-params.ts +0 -254
  553. package/src/train/loss.ts +0 -1591
  554. package/src/train/merge.ts +0 -195
  555. package/src/train/optimizer.ts +0 -183
  556. package/src/train/prefix-shared.ts +0 -490
  557. package/src/train/rank.ts +0 -137
  558. package/src/train/segmented.ts +0 -1828
  559. package/src/train/steel-qmm-header.ts +0 -1461
  560. package/src/train/trainer.ts +0 -1221
  561. package/src/train/watch.ts +0 -542
  562. package/src/vision/diffusion-vision.ts +0 -620
  563. package/src/vision/embedder.ts +0 -168
  564. package/src/vision/preprocess.ts +0 -247
  565. package/src/vision/prompt.ts +0 -373
  566. package/src/vision/qwen3vl-preprocess.ts +0 -409
  567. package/src/vision/qwen3vl-prompt.ts +0 -196
  568. package/src/vision/qwen3vl-tower.ts +0 -555
  569. package/src/vision/siglip.ts +0 -599
  570. package/src/vision/video-frames.ts +0 -148
  571. package/src/web/app.html +0 -2143
  572. package/src/web/app.js +0 -5703
  573. package/src/web/skills/memory/SKILL.md +0 -120
  574. package/src/web/skills/web-research/SKILL.md +0 -46
  575. package/src/web/skills.ts +0 -97
  576. package/src/web/src/adapters-panel.ts +0 -288
  577. package/src/web/src/assistant.ts +0 -451
  578. package/src/web/src/chat.ts +0 -1057
  579. package/src/web/src/composer.ts +0 -959
  580. package/src/web/src/dataset.ts +0 -132
  581. package/src/web/src/finetune.ts +0 -267
  582. package/src/web/src/hub.ts +0 -428
  583. package/src/web/src/main.ts +0 -98
  584. package/src/web/src/markdown.ts +0 -0
  585. package/src/web/src/memory-panel.ts +0 -572
  586. package/src/web/src/model-picker.ts +0 -159
  587. package/src/web/src/palette.ts +0 -346
  588. package/src/web/src/protocol.ts +0 -26
  589. package/src/web/src/quantize.ts +0 -167
  590. package/src/web/src/rag.ts +0 -274
  591. package/src/web/src/sessions.ts +0 -469
  592. package/src/web/src/shell.ts +0 -804
  593. package/src/web/src/ui-catalog.ts +0 -78
  594. package/src/web/vendor/README.md +0 -61
  595. package/src/web-assets.ts +0 -47
  596. package/src/web-tools.ts +0 -478
  597. package/src/weights.ts +0 -255
  598. /package/src/{serve → server}/completion-sink.ts +0 -0
  599. /package/src/web/{icon.svg → public/icon.svg} +0 -0
  600. /package/src/web/{manifest.webmanifest → public/manifest.webmanifest} +0 -0
  601. /package/src/web/{vendor → public/vendor}/hljs-LICENSE +0 -0
@@ -1,60 +0,0 @@
1
- // UltraFeedback (binarized) → mlx-bun preference rows. The binarized schema
2
- // (HuggingFaceH4/ultrafeedback_binarized) gives each row a `prompt` plus
3
- // `chosen` / `rejected` as message lists; ORPO wants {prompt, chosen, rejected}
4
- // where chosen/rejected are the assistant response *strings*. Pure transform so
5
- // it is unit-testable without downloading the dataset.
6
-
7
- export interface PreferenceRow {
8
- prompt: string;
9
- chosen: string;
10
- rejected: string;
11
- }
12
-
13
- type Msg = { role?: string; content?: string };
14
-
15
- /** Last assistant message content in a binarized message list (the response). */
16
- function lastAssistant(messages: unknown): string | null {
17
- if (!Array.isArray(messages)) return null;
18
- for (let i = messages.length - 1; i >= 0; i--) {
19
- const m = messages[i] as Msg;
20
- if (m && m.role === "assistant" && typeof m.content === "string" && m.content.trim())
21
- return m.content;
22
- }
23
- return null;
24
- }
25
-
26
- /** Convert one binarized UltraFeedback row → a preference row, or null if it is
27
- * unusable (missing fields, or chosen == rejected so there is no signal). */
28
- export function toPreferenceRow(row: Record<string, unknown>): PreferenceRow | null {
29
- const prompt = typeof row.prompt === "string" ? row.prompt : null;
30
- const chosen = lastAssistant(row.chosen);
31
- const rejected = lastAssistant(row.rejected);
32
- if (!prompt || !chosen || !rejected) return null;
33
- if (chosen.trim() === rejected.trim()) return null;
34
- return { prompt, chosen, rejected };
35
- }
36
-
37
- /** Rough token estimate (≈4 chars/token) for a cheap length pre-filter before
38
- * tokenization — keeps the curated set within a target context without loading
39
- * a tokenizer. */
40
- export function approxTokens(text: string): number {
41
- return Math.ceil(text.length / 4);
42
- }
43
-
44
- /** Filter + convert a stream of binarized rows. Drops unusable rows and ones
45
- * whose prompt+chosen or prompt+rejected approx-exceeds `maxApproxTokens`. */
46
- export function curate(
47
- rows: Record<string, unknown>[],
48
- maxApproxTokens = 2048,
49
- ): PreferenceRow[] {
50
- const out: PreferenceRow[] = [];
51
- for (const row of rows) {
52
- const p = toPreferenceRow(row);
53
- if (!p) continue;
54
- const base = approxTokens(p.prompt);
55
- if (base + approxTokens(p.chosen) > maxApproxTokens) continue;
56
- if (base + approxTokens(p.rejected) > maxApproxTokens) continue;
57
- out.push(p);
58
- }
59
- return out;
60
- }
package/src/evaldb.ts DELETED
@@ -1,199 +0,0 @@
1
- // Eval DB: every benchmark run recorded with model + config + commit
2
- // (bun:sqlite). The fit calculator's predictions are validated against
3
- // the measured peaks recorded here.
4
-
5
- import { Database } from "bun:sqlite";
6
- import { mkdirSync } from "node:fs";
7
-
8
- export const DEFAULT_EVAL_DB = `${process.env.HOME}/.cache/mlx-bun/evals.sqlite`;
9
-
10
- const SCHEMA = `
11
- CREATE TABLE IF NOT EXISTS runs (
12
- id INTEGER PRIMARY KEY AUTOINCREMENT,
13
- ts INTEGER NOT NULL,
14
- model_path TEXT NOT NULL,
15
- commit_sha TEXT,
16
- prompt_tokens INTEGER NOT NULL,
17
- cached_tokens INTEGER NOT NULL DEFAULT 0,
18
- generated_tokens INTEGER NOT NULL,
19
- prefill_tps REAL NOT NULL,
20
- decode_tps REAL NOT NULL,
21
- peak_bytes INTEGER NOT NULL,
22
- predicted_peak_bytes INTEGER,
23
- predicted_decode_tps REAL,
24
- notes TEXT
25
- );
26
-
27
- -- Phase 18 P1 — parallel-load harness. ONE row per configuration point
28
- -- (a concurrency level or an arrival-rate target), not per request: the
29
- -- single-stream runs table can't hold p50/p95 distributions or a
30
- -- concurrency axis. Same additive-migration discipline as runs (these
31
- -- are real measurements, never dropped). stack + machine_state mirror the
32
- -- runs columns so cross-stack load rows read the same way.
33
- CREATE TABLE IF NOT EXISTS load_runs (
34
- id INTEGER PRIMARY KEY AUTOINCREMENT,
35
- ts INTEGER NOT NULL,
36
- model_id TEXT NOT NULL,
37
- base_url TEXT NOT NULL,
38
- stack TEXT NOT NULL DEFAULT 'mlx-bun',
39
- commit_sha TEXT,
40
- mode TEXT NOT NULL, -- 'closed-loop' | 'open-loop'
41
- concurrency INTEGER NOT NULL, -- in-flight cap (closed) or requester count (open)
42
- target_rpm REAL, -- open-loop arrival target per requester (null = closed)
43
- max_tokens INTEGER NOT NULL,
44
- requests_ok INTEGER NOT NULL,
45
- requests_err INTEGER NOT NULL,
46
- duration_s REAL NOT NULL, -- wall time of the measured window
47
- ttft_p50_ms REAL NOT NULL,
48
- ttft_p95_ms REAL NOT NULL,
49
- e2e_p50_ms REAL NOT NULL,
50
- e2e_p95_ms REAL NOT NULL,
51
- e2e_p99_ms REAL NOT NULL,
52
- agg_tps REAL NOT NULL, -- generated tokens/s summed across streams
53
- per_req_tps REAL NOT NULL, -- median per-stream decode tok/s
54
- achieved_rpm REAL NOT NULL, -- completed requests/min over the window
55
- peak_bytes INTEGER, -- from /stats if exposed, else null
56
- notes TEXT,
57
- machine_state TEXT
58
- );
59
- `;
60
-
61
- export interface EvalRun {
62
- modelPath: string;
63
- commitSha?: string | null;
64
- promptTokens: number;
65
- cachedTokens?: number;
66
- generatedTokens: number;
67
- prefillTps: number;
68
- decodeTps: number;
69
- peakBytes: number;
70
- predictedPeakBytes?: number | null;
71
- predictedDecodeTps?: number | null;
72
- notes?: string | null;
73
- /** Which engine produced the number: mlx-bun (default) | mlx-lm | optiq. */
74
- stack?: string;
75
- /** JSON snapshot of preflight machine state (swap, free %, thermal). */
76
- machineState?: string | null;
77
- }
78
-
79
- /** One configuration point of a parallel-load sweep (Phase 18 P1). */
80
- export interface LoadRun {
81
- modelId: string;
82
- baseUrl: string;
83
- stack?: string;
84
- commitSha?: string | null;
85
- /** 'closed-loop' (fixed in-flight) or 'open-loop' (target arrival rate). */
86
- mode: "closed-loop" | "open-loop";
87
- /** In-flight cap (closed-loop) or requester count (open-loop). */
88
- concurrency: number;
89
- /** Per-requester target rpm (open-loop only; null for closed-loop). */
90
- targetRpm?: number | null;
91
- maxTokens: number;
92
- requestsOk: number;
93
- requestsErr: number;
94
- durationS: number;
95
- ttftP50Ms: number;
96
- ttftP95Ms: number;
97
- e2eP50Ms: number;
98
- e2eP95Ms: number;
99
- e2eP99Ms: number;
100
- /** Generated tokens/s summed across all streams in the window. */
101
- aggTps: number;
102
- /** Median per-stream decode tok/s. */
103
- perReqTps: number;
104
- /** Completed requests per minute over the measured window. */
105
- achievedRpm: number;
106
- /** Peak resident bytes sampled from /stats, if the server exposes it. */
107
- peakBytes?: number | null;
108
- notes?: string | null;
109
- machineState?: string | null;
110
- }
111
-
112
- export class EvalDB {
113
- readonly db: Database;
114
-
115
- constructor(dbPath: string = DEFAULT_EVAL_DB) {
116
- if (dbPath !== ":memory:") {
117
- try { mkdirSync(dbPath.slice(0, dbPath.lastIndexOf("/")), { recursive: true }); } catch {}
118
- }
119
- this.db = new Database(dbPath);
120
- this.db.exec(SCHEMA);
121
- // Additive migrations — eval rows are real data, never dropped
122
- // (unlike the registry, which is a rebuildable cache).
123
- const cols = (this.db.query("PRAGMA table_info(runs)").all() as { name: string }[])
124
- .map((c) => c.name);
125
- if (!cols.includes("stack"))
126
- this.db.exec("ALTER TABLE runs ADD COLUMN stack TEXT NOT NULL DEFAULT 'mlx-bun'");
127
- if (!cols.includes("machine_state"))
128
- this.db.exec("ALTER TABLE runs ADD COLUMN machine_state TEXT");
129
- }
130
-
131
- record(run: EvalRun): void {
132
- this.db.prepare(`
133
- INSERT INTO runs (ts, model_path, commit_sha, prompt_tokens, cached_tokens,
134
- generated_tokens, prefill_tps, decode_tps, peak_bytes,
135
- predicted_peak_bytes, predicted_decode_tps, notes, stack, machine_state)
136
- VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
137
- `).run(
138
- Date.now(), run.modelPath, run.commitSha ?? null,
139
- run.promptTokens, run.cachedTokens ?? 0, run.generatedTokens,
140
- run.prefillTps, run.decodeTps, run.peakBytes,
141
- run.predictedPeakBytes ?? null, run.predictedDecodeTps ?? null,
142
- run.notes ?? null, run.stack ?? "mlx-bun", run.machineState ?? null,
143
- );
144
- }
145
-
146
- /** Record one parallel-load configuration point (Phase 18 P1). */
147
- recordLoad(run: LoadRun): void {
148
- this.db.prepare(`
149
- INSERT INTO load_runs (ts, model_id, base_url, stack, commit_sha, mode,
150
- concurrency, target_rpm, max_tokens, requests_ok, requests_err,
151
- duration_s, ttft_p50_ms, ttft_p95_ms, e2e_p50_ms, e2e_p95_ms, e2e_p99_ms,
152
- agg_tps, per_req_tps, achieved_rpm, peak_bytes, notes, machine_state)
153
- VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
154
- `).run(
155
- Date.now(), run.modelId, run.baseUrl, run.stack ?? "mlx-bun",
156
- run.commitSha ?? null, run.mode, run.concurrency, run.targetRpm ?? null,
157
- run.maxTokens, run.requestsOk, run.requestsErr, run.durationS,
158
- run.ttftP50Ms, run.ttftP95Ms, run.e2eP50Ms, run.e2eP95Ms, run.e2eP99Ms,
159
- run.aggTps, run.perReqTps, run.achievedRpm, run.peakBytes ?? null,
160
- run.notes ?? null, run.machineState ?? null,
161
- );
162
- }
163
-
164
- recentLoad(limit = 20): Record<string, unknown>[] {
165
- return this.db
166
- .query("SELECT * FROM load_runs ORDER BY ts DESC LIMIT ?")
167
- .all(limit) as Record<string, unknown>[];
168
- }
169
-
170
- /** Latest mlx-bun measurement for a model snapshot (status page:
171
- * measured numbers beat predictions when available). */
172
- latestFor(modelPath: string): { decodeTps: number; ts: number; notes: string | null } | null {
173
- const r = this.db
174
- .query("SELECT decode_tps, ts, notes FROM runs WHERE model_path = ? AND stack = 'mlx-bun' ORDER BY ts DESC LIMIT 1")
175
- .get(modelPath) as { decode_tps: number; ts: number; notes: string | null } | null;
176
- return r ? { decodeTps: r.decode_tps, ts: r.ts, notes: r.notes } : null;
177
- }
178
-
179
- recent(limit = 20): Record<string, unknown>[] {
180
- return this.db
181
- .query("SELECT * FROM runs ORDER BY ts DESC LIMIT ?")
182
- .all(limit) as Record<string, unknown>[];
183
- }
184
-
185
- close(): void {
186
- this.db.close();
187
- }
188
- }
189
-
190
- export function gitCommit(): string | null {
191
- try {
192
- const r = Bun.spawnSync(["git", "rev-parse", "--short", "HEAD"], {
193
- cwd: import.meta.dir,
194
- });
195
- return r.exitCode === 0 ? r.stdout.toString().trim() : null;
196
- } catch {
197
- return null;
198
- }
199
- }
package/src/expert-io.ts DELETED
@@ -1,393 +0,0 @@
1
- import { dlopen, FFIType, ptr as ffiPtr, toArrayBuffer } from "bun:ffi";
2
- import { existsSync } from "node:fs";
3
- import { dirname, join } from "node:path";
4
- import { nativePackDir } from "./native-pack";
5
- import type { MlxArray } from "./mlx/array";
6
- import type { MlxHandle } from "./mlx/ffi";
7
- import { runtimeValue } from "./runtime-config";
8
-
9
- const { cstring, i32, ptr: pointer, u32, u64 } = FFIType;
10
- const EBUSY = 16;
11
- const EAGAIN = 35;
12
- const MAX_SEGMENTS = 8;
13
- const HINT_STATS = 8;
14
-
15
- type ExpertIOState = "open" | "closing" | "closed";
16
-
17
- /**
18
- * Poll an asynchronous expert read without letting an await cross the
19
- * store's lifetime boundary. Exported only so the race can be tested with a
20
- * deterministic fake poller rather than relying on native I/O timing.
21
- */
22
- export async function _pollExpertReadUntilReady(
23
- assertOpen: () => void,
24
- poll: () => number,
25
- sleep: () => Promise<void> = () => Bun.sleep(1),
26
- ): Promise<void> {
27
- for (;;) {
28
- assertOpen();
29
- const status = poll();
30
- if (status === 0) return;
31
- if (status !== EAGAIN) throw new Error(`expert read failed: errno ${status}`);
32
- await sleep();
33
- }
34
- }
35
-
36
- export interface ExpertIOOptions {
37
- slots: number;
38
- slotBytes: number;
39
- workers?: number;
40
- alignment?: number;
41
- noCache?: boolean;
42
- /** Keep loaded anonymous slot pages out of the macOS compressor. */
43
- wireSlots?: boolean;
44
- libraryPath?: string;
45
- }
46
-
47
- export interface ExpertIOSegment {
48
- readonly file: number;
49
- readonly offset: number;
50
- readonly destination: number;
51
- readonly length: number;
52
- }
53
-
54
- export interface ExpertIOHintSegment {
55
- readonly file: number;
56
- readonly offset: number;
57
- readonly length: number;
58
- }
59
-
60
- export interface ExpertIOHintSnapshot {
61
- readonly submitted: number;
62
- readonly completed: number;
63
- readonly dropped: number;
64
- readonly operations: number;
65
- readonly bytes: number;
66
- readonly errors: number;
67
- readonly queueDepth: number;
68
- readonly inFlight: number;
69
- }
70
-
71
- function resolveLibrary(explicit?: string): string {
72
- if (explicit) return explicit;
73
- const configured = runtimeValue("MLX_BUN_EXPERT_IO_DYLIB");
74
- if (configured) return configured;
75
- const candidates = [
76
- join(dirname(process.execPath), "libmlx_bun_expert_io.dylib"),
77
- join(nativePackDir(), "libmlx_bun_expert_io.dylib"),
78
- join(import.meta.dir, "..", "dist-native", "libmlx_bun_expert_io.dylib"),
79
- ];
80
- return candidates.find(existsSync) ?? candidates[0]!;
81
- }
82
-
83
- export class ExpertIOSlabStore {
84
- readonly files: readonly string[];
85
- readonly slots: number;
86
- readonly slotBytes: number;
87
- #lib: any;
88
- #handle: bigint;
89
- #state: ExpertIOState = "open";
90
- readonly #wireSlots: boolean;
91
-
92
- constructor(path: string | readonly string[], options: ExpertIOOptions) {
93
- if (!Number.isSafeInteger(options.slots) || options.slots <= 0) throw new Error("slots must be positive");
94
- if (!Number.isSafeInteger(options.slotBytes) || options.slotBytes <= 0) throw new Error("slotBytes must be positive");
95
- const paths = typeof path === "string" ? [path] : [...path];
96
- if (paths.length === 0 || paths.some((entry) => entry.length === 0))
97
- throw new Error("expert slab store requires at least one file");
98
- this.files = Object.freeze(paths);
99
- this.slots = options.slots; this.slotBytes = options.slotBytes;
100
- this.#wireSlots = options.wireSlots === true;
101
- this.#lib = dlopen(resolveLibrary(options.libraryPath), {
102
- mlx_bun_expert_io_open: { args: [cstring, u32, u64, u64, u32, i32], returns: u64 },
103
- mlx_bun_expert_io_open_many: {
104
- args: [pointer, u64, u32, u32, u64, u64, u32, i32],
105
- returns: u64,
106
- },
107
- mlx_bun_expert_io_submit: { args: [u64, u32, u64, u64, u64], returns: i32 },
108
- mlx_bun_expert_io_submitv: {
109
- args: [u64, u32, u64, pointer, pointer, pointer, pointer, u32],
110
- returns: i32,
111
- },
112
- mlx_bun_expert_io_hintv: {
113
- args: [u64, pointer, pointer, pointer, u32],
114
- returns: i32,
115
- },
116
- mlx_bun_expert_io_hint_stats: {
117
- args: [u64, pointer, u32],
118
- returns: i32,
119
- },
120
- mlx_bun_expert_io_wait: { args: [u64, u32, u64], returns: i32 },
121
- mlx_bun_expert_io_poll: { args: [u64, u32, u64], returns: i32 },
122
- mlx_bun_expert_io_cancel: { args: [u64, u32, u64], returns: i32 },
123
- mlx_bun_expert_io_lease: { args: [u64, u32, u64, i32], returns: i32 },
124
- mlx_bun_expert_io_release: { args: [u64, u32, u64, i32], returns: i32 },
125
- mlx_bun_expert_io_discard: { args: [u64, u32, u64], returns: i32 },
126
- mlx_bun_expert_io_set_wiring: { args: [u64, i32], returns: i32 },
127
- mlx_bun_process_phys_footprint: { args: [], returns: u64 },
128
- mlx_bun_process_compressed: { args: [], returns: u64 },
129
- mlx_bun_expert_io_ptr: { args: [u64, u32, u64], returns: u64 },
130
- mlx_bun_expert_io_length: { args: [u64, u32, u64], returns: u64 },
131
- mlx_bun_expert_io_close: { args: [u64], returns: i32 },
132
- });
133
- const encoded = Buffer.from(`${paths.join("\0")}\0`);
134
- this.#handle = this.#lib.symbols.mlx_bun_expert_io_open_many(
135
- ffiPtr(encoded), BigInt(encoded.byteLength), paths.length,
136
- options.slots, BigInt(options.slotBytes),
137
- BigInt(options.alignment ?? 16 * 1024),
138
- options.workers ?? 2, options.noCache === false ? 0 : 1,
139
- ) as bigint;
140
- if (this.#handle === 0n) {
141
- this.#lib.close();
142
- throw new Error(`failed to open expert slab store: ${paths.join(", ")}`);
143
- }
144
- if (this.#wireSlots) {
145
- const status = this.#lib.symbols.mlx_bun_expert_io_set_wiring(
146
- this.#handle,
147
- 1,
148
- );
149
- if (status !== 0) {
150
- this.#lib.symbols.mlx_bun_expert_io_close(this.#handle);
151
- this.#lib.close();
152
- throw new Error(`failed to enable expert slot wiring: errno ${status}`);
153
- }
154
- }
155
- }
156
-
157
- submit(slot: number, generation: bigint, offset: number, length: number): void {
158
- this.submitSegments(slot, generation, [{
159
- file: 0,
160
- offset,
161
- destination: 0,
162
- length,
163
- }]);
164
- }
165
-
166
- submitSegments(
167
- slot: number,
168
- generation: bigint,
169
- segments: readonly ExpertIOSegment[],
170
- ): void {
171
- this.#checkSlot(slot);
172
- if (generation <= 0n) throw new RangeError("expert generation must be positive");
173
- if (segments.length === 0 || segments.length > MAX_SEGMENTS)
174
- throw new RangeError(`expert read requires 1..${MAX_SEGMENTS} segments`);
175
- const files = new Uint32Array(segments.length);
176
- const offsets = new BigUint64Array(segments.length);
177
- const destinations = new BigUint64Array(segments.length);
178
- const lengths = new BigUint64Array(segments.length);
179
- let extent = 0;
180
- for (let index = 0; index < segments.length; index++) {
181
- const segment = segments[index]!;
182
- if (!Number.isSafeInteger(segment.file) || segment.file < 0 ||
183
- segment.file >= this.files.length)
184
- throw new RangeError(`expert segment ${index} file is out of range`);
185
- if (!Number.isSafeInteger(segment.offset) || segment.offset < 0)
186
- throw new RangeError(`expert segment ${index} offset must be non-negative`);
187
- if (!Number.isSafeInteger(segment.destination) || segment.destination < 0)
188
- throw new RangeError(`expert segment ${index} destination must be non-negative`);
189
- if (!Number.isSafeInteger(segment.length) || segment.length <= 0)
190
- throw new RangeError(`expert segment ${index} length must be positive`);
191
- const end = segment.destination + segment.length;
192
- if (!Number.isSafeInteger(end) || end > this.slotBytes)
193
- throw new RangeError(`expert segment ${index} exceeds slot capacity`);
194
- for (let prior = 0; prior < index; prior++) {
195
- const other = segments[prior]!;
196
- const otherEnd = other.destination + other.length;
197
- if (segment.destination < otherEnd && other.destination < end)
198
- throw new RangeError(`expert segment ${index} overlaps segment ${prior}`);
199
- }
200
- files[index] = segment.file;
201
- offsets[index] = BigInt(segment.offset);
202
- destinations[index] = BigInt(segment.destination);
203
- lengths[index] = BigInt(segment.length);
204
- extent = Math.max(extent, end);
205
- }
206
- if (extent > this.slotBytes)
207
- throw new RangeError("expert segmented read exceeds slot capacity");
208
- const status = this.#lib.symbols.mlx_bun_expert_io_submitv(
209
- this.#handle,
210
- slot,
211
- generation,
212
- ffiPtr(files),
213
- ffiPtr(offsets),
214
- ffiPtr(destinations),
215
- ffiPtr(lengths),
216
- segments.length,
217
- );
218
- if (status !== 0)
219
- throw new Error(`expert segmented read submit failed: errno ${status}`);
220
- }
221
-
222
- async wait(slot: number, generation: bigint): Promise<void> {
223
- await _pollExpertReadUntilReady(
224
- () => this.#checkSlot(slot),
225
- () => this.#lib.symbols.mlx_bun_expert_io_poll(this.#handle, slot, generation),
226
- );
227
- }
228
-
229
- /** Queue advisory readahead on the dedicated hint worker. No slab is used. */
230
- hintSegments(segments: readonly ExpertIOHintSegment[]): boolean {
231
- this.#checkOpen();
232
- if (segments.length === 0 || segments.length > MAX_SEGMENTS)
233
- throw new RangeError(`expert hint requires 1..${MAX_SEGMENTS} segments`);
234
- const files = new Uint32Array(segments.length);
235
- const offsets = new BigUint64Array(segments.length);
236
- const lengths = new BigUint64Array(segments.length);
237
- for (let index = 0; index < segments.length; index++) {
238
- const segment = segments[index]!;
239
- if (!Number.isSafeInteger(segment.file) || segment.file < 0 ||
240
- segment.file >= this.files.length)
241
- throw new RangeError(`expert hint ${index} file is out of range`);
242
- if (!Number.isSafeInteger(segment.offset) || segment.offset < 0)
243
- throw new RangeError(`expert hint ${index} offset must be non-negative`);
244
- if (!Number.isSafeInteger(segment.length) || segment.length <= 0)
245
- throw new RangeError(`expert hint ${index} length must be positive`);
246
- files[index] = segment.file;
247
- offsets[index] = BigInt(segment.offset);
248
- lengths[index] = BigInt(segment.length);
249
- }
250
- const status = this.#lib.symbols.mlx_bun_expert_io_hintv(
251
- this.#handle,
252
- ffiPtr(files),
253
- ffiPtr(offsets),
254
- ffiPtr(lengths),
255
- segments.length,
256
- );
257
- if (status === 0) return true;
258
- if (status === EAGAIN) return false;
259
- throw new Error(`expert hint submit failed: errno ${status}`);
260
- }
261
-
262
- hintTelemetry(): ExpertIOHintSnapshot {
263
- this.#checkOpen();
264
- const out = new BigUint64Array(HINT_STATS);
265
- const status = this.#lib.symbols.mlx_bun_expert_io_hint_stats(
266
- this.#handle,
267
- ffiPtr(out),
268
- out.length,
269
- );
270
- if (status !== 0)
271
- throw new Error(`expert hint telemetry failed: errno ${status}`);
272
- return {
273
- submitted: Number(out[0]),
274
- completed: Number(out[1]),
275
- dropped: Number(out[2]),
276
- operations: Number(out[3]),
277
- bytes: Number(out[4]),
278
- errors: Number(out[5]),
279
- queueDepth: Number(out[6]),
280
- inFlight: Number(out[7]),
281
- };
282
- }
283
-
284
- cancel(slot: number, generation: bigint): void {
285
- this.#checkSlot(slot);
286
- const status = this.#lib.symbols.mlx_bun_expert_io_cancel(this.#handle, slot, generation);
287
- if (status !== 0) throw new Error(`expert read cancel failed: errno ${status}`);
288
- }
289
-
290
- lease(slot: number, generation: bigint, consumer: "cpu" | "gpu" = "cpu"): void {
291
- this.#checkSlot(slot);
292
- const status = this.#lib.symbols.mlx_bun_expert_io_lease(this.#handle, slot, generation, consumer === "gpu" ? 1 : 0);
293
- if (status !== 0) throw new Error(`expert slot lease failed: errno ${status}`);
294
- }
295
-
296
- releaseCpu(slot: number, generation: bigint): void {
297
- this.#checkSlot(slot);
298
- const status = this.#lib.symbols.mlx_bun_expert_io_release(this.#handle, slot, generation, 0);
299
- if (status !== 0) throw new Error(`expert slot release failed: errno ${status}`);
300
- }
301
-
302
- releaseGpuFenced(slot: number, generation: bigint): void {
303
- this.#checkSlot(slot);
304
- const status = this.#lib.symbols.mlx_bun_expert_io_release(this.#handle, slot, generation, 1);
305
- if (status !== 0) throw new Error(`expert GPU slot release failed: errno ${status}`);
306
- }
307
-
308
- async releaseAfterGpuSync(
309
- slot: number, generation: bigint, dependentOutputs: readonly MlxArray[], stream?: MlxHandle,
310
- ): Promise<void> {
311
- this.#checkSlot(slot);
312
- if (dependentOutputs.length === 0) throw new Error("GPU slot release requires dependent outputs to evaluate");
313
- const [{ gpuStream }, { synchronize }, ops] = await Promise.all([
314
- import("./mlx/array"), import("./mlx/ffi"), import("./mlx/ops"),
315
- ]);
316
- ops.evalAll([...dependentOutputs]);
317
- synchronize(stream ?? gpuStream);
318
- this.releaseGpuFenced(slot, generation);
319
- }
320
-
321
- discard(slot: number, generation: bigint): void {
322
- this.#checkSlot(slot);
323
- const status = this.#lib.symbols.mlx_bun_expert_io_discard(this.#handle, slot, generation);
324
- if (status !== 0) throw new Error(`expert slot discard failed: errno ${status}`);
325
- }
326
-
327
- physicalFootprint(): number {
328
- this.#checkOpen();
329
- return Number(this.#lib.symbols.mlx_bun_process_phys_footprint());
330
- }
331
-
332
- compressedMemory(): number {
333
- this.#checkOpen();
334
- return Number(this.#lib.symbols.mlx_bun_process_compressed());
335
- }
336
-
337
- pointer(slot: number, generation: bigint): number {
338
- this.#checkSlot(slot);
339
- const result = Number(this.#lib.symbols.mlx_bun_expert_io_ptr(this.#handle, slot, generation));
340
- if (!result) throw new Error("expert slot pointer requires the matching active lease");
341
- return result;
342
- }
343
-
344
- view(slot: number, generation: bigint, length?: number): Uint8Array {
345
- this.#checkSlot(slot);
346
- const resolvedLength = length ?? Number(this.#lib.symbols.mlx_bun_expert_io_length(this.#handle, slot, generation));
347
- if (resolvedLength < 0 || resolvedLength > this.slotBytes) throw new RangeError("expert slot view exceeds capacity");
348
- return new Uint8Array(toArrayBuffer(this.pointer(slot, generation) as never, 0, resolvedLength));
349
- }
350
-
351
- close(): void {
352
- if (this.#state === "closed") return;
353
- if (this.#state === "closing") return;
354
- this.#state = "closing";
355
- let status: number;
356
- try {
357
- status = this.#lib.symbols.mlx_bun_expert_io_close(this.#handle);
358
- } catch (error) {
359
- this.#state = "open";
360
- throw error;
361
- }
362
- if (status !== 0) {
363
- this.#state = "open";
364
- throw new Error(`expert slab close failed: errno ${status}`);
365
- }
366
- this.#state = "closed";
367
- this.#lib.close();
368
- }
369
-
370
- #checkSlot(slot: number): void {
371
- this.#checkOpen();
372
- if (!Number.isInteger(slot) || slot < 0 || slot >= this.slots) throw new RangeError("expert slot out of range");
373
- }
374
-
375
- #checkOpen(): void {
376
- if (this.#state !== "open") throw new Error("expert slab store is closed");
377
- }
378
- }
379
-
380
- export { EBUSY as EXPERT_IO_BUSY_ERRNO };
381
-
382
- export interface LruCandidate { slot: number; lastUse: bigint; pinned?: boolean; leased?: boolean }
383
-
384
- /** Deterministic eviction primitive: oldest use wins, then lowest slot ID. */
385
- export function selectExpertLruVictim(candidates: readonly LruCandidate[]): number | null {
386
- let best: LruCandidate | null = null;
387
- for (const candidate of candidates) {
388
- if (candidate.pinned || candidate.leased) continue;
389
- if (!best || candidate.lastUse < best.lastUse ||
390
- (candidate.lastUse === best.lastUse && candidate.slot < best.slot)) best = candidate;
391
- }
392
- return best?.slot ?? null;
393
- }