Onnx/WebGPU Support

Onnx/WebGPU Support#

This section is a process guide for the MLX -> Graph IR -> ONNX -> WebGPU path in MLX Ruby.

If you want API-by-API reference details, see Export Functions.

Architecture boundary (post-refactor)#

Use MLX::ONNX.* methods as the public API. The implementation is split across:

  • MLX::ONNX

  • MLX::ONNX::Native

  • MLX::ONNX::WebGPUHarness

The ONNX lowering core now lives in the separate mlx-onnx repository and is consumed here via submodules/mlx-onnx.

Pipeline stages#

  1. Capture MLX execution as Graph IR.

  2. Convert Graph IR to ONNX (JSON stub and/or binary model).

  3. Run optional compatibility diagnostics (native report).

  4. Package browser harness assets.

  5. Run smoke validation in Node/browser runtime.

Pages in this section:

Contents