solprop.backends.mlx_gpu#

The mlx backend: single-precision propagation on Apple-silicon GPUs (SEP 33).

Registered as "mlx" and installed with the mlx extra on macOS on Apple silicon. It runs precision="single" problems only, validated against solprop.backends.numpy_ref.single, its oracle (ADR-054). The package is named mlx_gpu so that it does not shadow the mlx library inside it.

Importing it sets MLX_ENABLE_TF32=0 when the variable is unset, so that MLX computes float32 matrix products in float32 rather than TF32; MLX reads the setting when its subsystem first initialises, so a process that used MLX before is checked by gemm_probe() and refused if its products are TF32.

Modules

arithmetic

The protected operations of the mlx backend, and the probes that establish them.

backend

The mlx backend: the single-precision oracle's step, on the Metal GPU.

batch

Fixed-schedule lockstep batches of mlx members (SEP 33 section 6.3, Q10).

envelope

The envelope-grid plasma on the Metal device (SEP 33 section 6.3, phase 4).

passes

The passes of one stage on the device, in binary32: the oracle's, in MLX.