mlxsmith 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

mlxsmith/accel/__init__.py

@@ -1,10 +1,7 @@
 from __future__ import annotations
 from .none import NoneBackend
-from .zmlx_backend import ZMLXBackend
 
 def get_backend(name: str):
     if name == "none":
         return NoneBackend()
-    if name == "zmlx":
-        return ZMLXBackend()
     raise ValueError(f"Unknown accel backend: {name}")

mlxsmith/cli.py

@@ -83,7 +83,6 @@ def doctor():
     table.add_row("cpu_count", str(info.cpu_count))
     table.add_row("metal", str(info.has_metal))
     table.add_row("mlx", f"{info.has_mlx} {info.mlx_version or ''}".strip())
-    table.add_row("zmlx", str(info.has_zmlx))
     console.print(table)
 
 
@@ -729,7 +728,7 @@ def rlm_history(limit: int = typer.Option(10, "--limit")):
 
 @accel_app.command("status")
 def accel_status():
-    backends = ["none", "zmlx"]
+    backends = ["none"]
     table = Table(title="mlxsmith accel status")
     table.add_column("backend")
     table.add_column("available")

mlxsmith/config_models.py

@@ -6,7 +6,7 @@ from typing import Dict, List, Literal, Optional, Any
 
 from pydantic import BaseModel, Field, field_validator
 
-AccelBackendName = Literal["none", "zmlx"]
+AccelBackendName = Literal["none"]
 
 
 class ModelConfig(BaseModel):

mlxsmith/util.py

@@ -46,7 +46,6 @@ class SystemInfo:
     has_metal: Optional[bool]
     has_mlx: bool
     mlx_version: Optional[str]
-    has_zmlx: bool
 
 def detect_system() -> SystemInfo:
     has_mlx = False
@@ -58,10 +57,6 @@ def detect_system() -> SystemInfo:
     except Exception:
         pass
 
-    import importlib.util
-
-    has_zmlx = importlib.util.find_spec("zmlx") is not None
-
     # Metal detection (best-effort): on macOS we assume Metal is present; for CI, this is not reliable.
     has_metal = None
     if sys.platform == "darwin":
@@ -83,7 +78,6 @@ def detect_system() -> SystemInfo:
         has_metal=has_metal,
         has_mlx=has_mlx,
         mlx_version=mlx_version,
-        has_zmlx=has_zmlx,
     )
 
 def require(cond: bool, msg: str):

{mlxsmith-0.1.1.dist-info → mlxsmith-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mlxsmith
-Version: 0.1.1
+Version: 0.1.2
 Summary: Apple Silicon MLX fine-tuning toolkit — SFT, DPO/ORPO, GRPO, distillation, and OpenAI-compatible serving.
 Author-email: Shannon Labs <hmbown@gmail.com>
 License: MIT
@@ -40,8 +40,6 @@ Provides-Extra: serve
 Requires-Dist: fastapi>=0.128.0; extra == "serve"
 Requires-Dist: uvicorn>=0.40.0; extra == "serve"
 Requires-Dist: httpx>=0.28.0; extra == "serve"
-Provides-Extra: zmlx
-Requires-Dist: zmlx; extra == "zmlx"
 Provides-Extra: dev
 Requires-Dist: pytest>=9.0.0; extra == "dev"
 Requires-Dist: ruff>=0.14.0; extra == "dev"
@@ -59,7 +57,7 @@ Dynamic: license-file
 
 Apple Silicon MLX fine-tuning toolkit — SFT, DPO/ORPO, GRPO, distillation, and OpenAI-compatible serving.
 
-**Status:** alpha (v0.1.0). Full training pipeline validated on Qwen3-4B.
+**Status:** alpha (v0.1.2). Full training pipeline validated on Qwen3-4B.
 
 ## Install
 
@@ -85,7 +83,7 @@ pip install "mlxsmith[all]"
 ```bash
 mlxsmith init myproj
 cd myproj
-mlxsmith doctor        # check Python, MLX, Metal, ZMLX
+mlxsmith doctor        # check Python, MLX, Metal
 ```
 
 ## Training
@@ -269,14 +267,6 @@ mlxsmith rlm history               # view history
 
 Includes task generation, mutation for data diversity, corpus management, EMA-based gating, and weight pointer IPC for multi-process coordination. See `docs/orchestrator.md`.
 
-### ZMLX acceleration
-
-Optional zero-copy MLX acceleration backend.
-
-```bash
-mlxsmith accel status
-```
-
 ## Docs
 
 - `docs/PROJECT_FORMAT.md` — project layout and artifacts

{mlxsmith-0.1.1.dist-info → mlxsmith-0.1.2.dist-info}/RECORD

@@ -2,20 +2,19 @@ mlxsmith/__init__.py,sha256=CJZKl9Hp16DYlQR7yqstir-cL4n7GCw90d-meXliCHk,48
 mlxsmith/adapters.py,sha256=wkQ2q_ugaxCviNARSmxehwBcc2_NKVJ7mOofT-y30TY,1318
 mlxsmith/auth.py,sha256=_j_gx5ccZfpHs0_Xmpcgh_ELhX3ZBJLg2YYpjA-aPRI,2195
 mlxsmith/bench.py,sha256=VBgY9uOGm3xhmL3UrNNOnUoa3P0yaVQQ7wxykIEmDEw,3778
-mlxsmith/cli.py,sha256=I7vzO2Z3ja1ncMOGHhXSEN_kElfWdmA6Jvs4rFgINZ0,35767
+mlxsmith/cli.py,sha256=MAS21bSz4DTkdMAao99KdtI0FtZFUEVQatSsS7VHPQ0,35713
 mlxsmith/config.py,sha256=K1VbN-3WoWf4vzZ6BNeTgEz2DFH8s9YminqwyNBVLj0,16389
-mlxsmith/config_models.py,sha256=pMsLGyC9J9T9jqs5ipYFUuEkfir7iiDA4EZdcb65k5g,8407
+mlxsmith/config_models.py,sha256=x827pTs-SfPyqXI3mPR6h-cz7KQ69zQICPAaeGlW-A4,8399
 mlxsmith/data.py,sha256=3ZlNS8bnD7LlWACEmULbf8RGQzCuf0QulFpI1PWvNuI,16160
 mlxsmith/eval.py,sha256=nSARSEKKwZM8Ot5rUDDpGikaClGNxvg0ifgGkTA6mM0,3792
 mlxsmith/infer.py,sha256=ekpHhTird0dnTJzFOc-O98rjwkEKgAr9AFicKlaB3MA,4610
 mlxsmith/models.py,sha256=BRaPTxzqy-5KEKdccveMgjpbRP0ZmbRnA_su8rz2P4k,8033
 mlxsmith/runs.py,sha256=2voYBryGGMlAKskHJ7TDiIPQL2_fFxSQ8RgtfGZ7ccg,1409
 mlxsmith/server.py,sha256=Fk-i9xK_Teq0Z0m-W1GRJVtcG0nYvd4bQ85lnAUuT1w,10690
-mlxsmith/util.py,sha256=8fagFtMP2YS1qlWkCt8bPWAz2jAgfvAf1ipNGJZAgIk,4544
-mlxsmith/accel/__init__.py,sha256=Mv2mj-2bLqUILhMUCjMLu7JORcu0-cGBOri32j7O5Xo,291
+mlxsmith/util.py,sha256=Cke2FxIVNmvpW9ElPxL5bc0JO_YFVHWtBwpR3QRtfWQ,4410
+mlxsmith/accel/__init__.py,sha256=fBY65q66jql1q5YMT9aIo96dgvVLHqS_LoJbVNA2xHY,201
 mlxsmith/accel/base.py,sha256=o0kmxV68KbxOmucC3eDjKbFA8dfTT2ETqN0XD_l3mM0,435
 mlxsmith/accel/none.py,sha256=WhxECIBv-pE63Vh1Iv86ObgT9JHOi4hA4BUyJc__sKU,362
-mlxsmith/accel/zmlx_backend.py,sha256=JfzQ44v9hrCJgcqU018ZD7qLNlubIe09CwYRpKyfwR8,1529
 mlxsmith/api/__init__.py,sha256=IrpIXDkUJm4BZqahYOK_0BkxvomlscEvCoLCm4GDxo8,998
 mlxsmith/api/handlers.py,sha256=94Spq4glFp7mRwmKrFqt7erd1Af_PxVP_vpxCo2UFdQ,46896
 mlxsmith/api/schemas.py,sha256=Q18kF9FKtvT1vdnXy6feSNTtCV2FiRWDzfdsPzc0te8,19316
@@ -61,9 +60,9 @@ mlxsmith/verifiers/jsonschema.py,sha256=hG_8c07Hwv-tpN2g0oxELwmLRxS8QGzRFwabmo4y
 mlxsmith/verifiers/pytest_verifier.py,sha256=ARNajzxUPNwtzSow6I2d0mLopZyvY29_d3F1sYVwEUY,2514
 mlxsmith/verifiers/regex.py,sha256=N7z3koE8Iy-a4DBs4404iQCNX2WGxequm5g4umric2Y,524
 mlxsmith/verifiers/types.py,sha256=FytBxB1OnNX1EcqZXSSs3WvL0GRv7byW4mfBMf6xP68,240
-mlxsmith-0.1.1.dist-info/licenses/LICENSE,sha256=ESYyLizI0WWtxMeS7rGVcX3ivMezm-HOd5WdeOh-9oU,1056
-mlxsmith-0.1.1.dist-info/METADATA,sha256=d0H3FBZa2-BfGVbz3TWpjtgWhMe-t46bp8PQELh5yFQ,9131
-mlxsmith-0.1.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-mlxsmith-0.1.1.dist-info/entry_points.txt,sha256=ys7GcKEjhzhkkTMBmmaNavTgsjqOuFnCKIG2w8Wcn6w,46
-mlxsmith-0.1.1.dist-info/top_level.txt,sha256=hKBwc8bn7uoI-_5Yhcq1T3IuChFhUFdzItIkZK1up6A,9
-mlxsmith-0.1.1.dist-info/RECORD,,
+mlxsmith-0.1.2.dist-info/licenses/LICENSE,sha256=ESYyLizI0WWtxMeS7rGVcX3ivMezm-HOd5WdeOh-9oU,1056
+mlxsmith-0.1.2.dist-info/METADATA,sha256=mtwo3IlnvH9ffCKWSHbxRMTLQFyzrvDi29BCyzbteqk,8963
+mlxsmith-0.1.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+mlxsmith-0.1.2.dist-info/entry_points.txt,sha256=ys7GcKEjhzhkkTMBmmaNavTgsjqOuFnCKIG2w8Wcn6w,46
+mlxsmith-0.1.2.dist-info/top_level.txt,sha256=hKBwc8bn7uoI-_5Yhcq1T3IuChFhUFdzItIkZK1up6A,9
+mlxsmith-0.1.2.dist-info/RECORD,,

mlxsmith/accel/zmlx_backend.py

@@ -1,42 +0,0 @@
-from __future__ import annotations
-
-from typing import Any, Dict
-from .base import AccelStats
-
-class ZMLXBackend:
-    name = "zmlx"
-
-    def __init__(self):
-        self._available = False
-        self._notes = {}
-        try:
-            import zmlx  # type: ignore
-            self._available = True
-            self._notes["zmlx_version"] = getattr(zmlx, "__version__", None)
-        except Exception as e:
-            self._available = False
-            self._notes["error"] = f"{type(e).__name__}: {e}"
-
-    def patch(self) -> None:
-        if not self._available:
-            # soft fail; caller should report status
-            return
-        # ZMLX can patch ops/modules. We keep this intentionally minimal and safe.
-        try:
-            import zmlx  # type: ignore
-            # If ZMLX provides a global patch hook, call it; otherwise, no-op.
-            patch_fn = getattr(zmlx, "patch", None)
-            if callable(patch_fn):
-                patch_fn()
-                self._notes["patched"] = True
-            else:
-                self._notes["patched"] = False
-                self._notes["hint"] = "No zmlx.patch() found; implement patch hook or integrate per-module."
-        except Exception as e:
-            self._notes["patched_error"] = f"{type(e).__name__}: {e}"
-
-    def warmup(self, model: Any, example_batch: Any) -> Dict[str, Any]:
-        return {"warmup": "not_implemented", "notes": self._notes}
-
-    def stats(self) -> AccelStats:
-        return AccelStats(backend="zmlx", notes=self._notes)