Skip to main content

Crate mlx_cpu

Crate mlx_cpu 

Source
Expand description

Pure Rust CPU backend — correctness oracle for conformance testing.

This crate re-exports the built-in CPU reference backend from mlx-core and will later provide optimized kernels (SIMD, rayon parallelism) as an upgrade path.

Structs§

CpuRefBackend
Reference CPU backend.

Functions§

cpu_stream
Create a new mlx_core::backend::Stream backed by the CPU reference backend.