elixir-nx / nx

Multi-dimensional arrays (tensors) and numerical definitions for Elixir
2.62k stars 190 forks source link

Investigate IREE integration #1481

Closed josevalim closed 2 months ago

josevalim commented 4 months ago

Unfortunately it does not support infeed/outfeed, so we need to make sure we raise accordingly.

polvalente commented 4 months ago

Work in progress at: pv-feat/iree-compiler

polvalente commented 2 months ago

https://github.com/elixir-nx/nx_iree is where this is going :)