Welcome!
Try running this example: nx/exla/examples/mnist.exs at main · elixir-nx/nx · GitHub
You will also need to set the XLA_TARGET environment variable accordingly: GitHub - elixir-nx/xla: Pre-compiled XLA extension · GitHub
Welcome!
Try running this example: nx/exla/examples/mnist.exs at main · elixir-nx/nx · GitHub
You will also need to set the XLA_TARGET environment variable accordingly: GitHub - elixir-nx/xla: Pre-compiled XLA extension · GitHub