Skip to contents

Fit an MLX linear model from design matrices

Source: R/mlxs-lm.R
mlxs_lm_fit.Rd

mlxs_lm_fit() powers mlxs_lm() by wrapping the QR-based solver that runs entirely on MLX arrays.

Usage

mlxs_lm_fit(x, y, weights = NULL)

Arguments

x

MLX design matrix (or object coercible via Rmlx::as_mlx()) whose rows represent observations and columns represent predictors.

y

MLX column vector (or object coercible via Rmlx::as_mlx()) holding the response values.

weights

Optional MLX column vector or numeric vector of non-negative observation weights. When supplied, weighted least squares are fit via the standard square-root weighting.

Value

A list with components coefficients, fitted.values, residuals, effects, and qr, mirroring the corresponding pieces of stats::lm(). Array-valued components remain MLX matrices to keep downstream GPU pipelines in device memory.

Details

Inputs that are not already MLX objects are converted with Rmlx::as_mlx() or Rmlx::mlx_matrix() so callers can provide base-R matrices or vectors. Weighted fits are performed by applying the standard square-root weight transform before solving the QR system.

Examples

x <- Rmlx::as_mlx(cbind(1, as.matrix(mtcars[c("cyl", "disp")])))
y <- Rmlx::mlx_matrix(mtcars$mpg, ncol = 1)
fit <- mlxs_lm_fit(x, y)
drop(as.matrix(fit$coefficients))
#> [1] 34.66099167 -1.58727658 -0.02058364