search

elixir-nx / nx

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

Investigate jax-metal integration #1480

Open josevalim opened 6 months ago

josevalim commented 6 months ago

Initial attempt here: https://github.com/elixir-nx/nx/pull/1504