Sounds like the Nx is running on the default binary backend which is native Elixir and very slow. Check your config.exs has:
config :nx,
default_backend: EXLA.Backend
Sounds like the Nx is running on the default binary backend which is native Elixir and very slow. Check your config.exs has:
config :nx,
default_backend: EXLA.Backend