onnxruntime-ep-mlx 0.31.1

MLX-native ONNX Runtime execution provider (plugin EP) for Apple Silicon — binds mlx-c directly, no mlx-rs.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

//! Wave-1 op handler modules. Each op = translate handler + claim predicate + registry entry.

pub mod attention;
pub mod attention_vjp;
pub mod broadcast_adjoint;
pub mod conv;
pub mod elementary_contrib;
pub mod elementwise;
mod host_scan;
pub mod image;
mod kv_cache;
pub(crate) mod lut4;
pub mod math;
pub mod matmul;
pub mod nonzero;
pub mod norm;
pub mod onnx_ml_linear;
pub mod onnx_ml_preprocess;
mod paged_attention;
pub mod paged_ssm;
pub(crate) mod qdq;
pub mod quant;
pub mod reduction;
pub mod shape;
// signal/random/recurrent/ssm/misc/controlflow
pub mod controlflow;
pub mod gated_delta_scan;
pub mod misc;
pub mod random;
pub mod recurrent;
pub mod selective_scan;
pub mod signal;
pub mod ssm;
pub mod stragglers;
pub mod unique;
pub mod vision;