mlx-lstsq 0.1.0

Least-squares solvers for MLX with Apple MPS native extensions.

mlx-lstsq

mlx-lstsq provides least-squares solvers for MLX backed by custom Apple Metal Performance Shaders kernels.

Requirements

• macOS on Apple Silicon
• Python 3.12 or newer
• mlx>=0.31.1
• Xcode command line tools

Installation

pip install mlx-lstsq

The package contains a compiled extension module plus companion .dylib and .metallib assets, so installation happens from a wheel on supported systems or from source with a local toolchain.

Usage

import mlx.core as mx
import mlx_lstsq

A = mx.array([[1.0, 0.0], [1.0, 1.0], [1.0, 2.0]], dtype=mx.float32)
b = mx.array([1.0, 2.0, 2.5], dtype=mx.float32)

x = mlx_lstsq.solve(A, b)
ridge_x = mlx_lstsq.solve_ridge(A, b, 1e-3)

Publishing Checklist

python3 -m pip install --upgrade build twine
python3 -m build
python3 -m twine check dist/*
python3 -m unittest discover -s tests -v

The smoke test installs the wheel from dist/ into a fresh virtual environment and verifies a scalar solve, so it checks the publishable artifact instead of the source tree. Run that check on each supported Python version before publishing.