Test elixir
Mix.install(
[
{:nx , "~> 0.12.1"},
{:exla, "~> 0.12.0"},
],
config: [
nx: [
default_backend: EXLA.Backend,
default_defn_options: [compiler: EXLA]
],
exla: [clients: [cuda: [platform: :cuda, preallocate: false]]],
]
)
Section
Nx.iota({2, 2})