NxIREE

Companion library to EXLA, providing bindings for the IREE runtime for MLIR.

MLIR modules can be obtained from Nx functions by calling EXLA.to_mlir_module/2 on them.

Installation

If available in Hex, the package can be installed by adding nx_iree to your list of dependencies in mix.exs:

def deps do
  [
    {:nx_iree, "~> 0.1.0"}
  ]
end

Documentation can be generated with ExDoc and published on HexDocs. Once published, the docs can be found at https://hexdocs.pm/nx_iree.