mlx-core 0.1.0

Safe Rust API for MLX tensors, devices, and core operations
Documentation
  • Coverage
  • 56.9%
    99 out of 174 items documented0 out of 83 items with examples
  • Size
  • Source code size: 111.66 kB This is the summed size of all the files inside the crates.io package for this release.
  • Documentation size: 8.06 MB This is the summed size of all files generated by rustdoc for all configured targets
  • Ø build duration
  • this release: 17s Average build duration of successful builds.
  • all releases: 17s Average build duration of successful builds in releases after 2024-10-23.
  • Links
  • crates.io
  • Dependencies
  • Versions
  • Owners
  • community-stevedores-org

Safe, ergonomic Rust API for MLX tensors and operations.

mlx-core provides the foundational types (Tensor, Device, DType, Shape) and a backend-agnostic interface for lazy tensor computation.

Architecture

Operations on tensors build a lazy computation graph. Calling eval() (or to_vec_f32()) triggers a topological walk that dispatches each node to the active Backend. A default CPU reference backend is provided out of the box.

Backends

  • Built-in CPU reference: simple, safe Rust — always available
  • ffi feature: delegates to the MLX C++ runtime via mlx-sys
  • mlx-cpu crate: optimized pure-Rust CPU backend (future)
  • mlx-metal crate: native Apple Silicon acceleration (future)