Teste

Mix.install([
  {:axon, "~> 0.7.0"}
])

Section

{backend, _rest} = Nx.default_backend()

IO.inspect backend

inputs = Nx.iota({2, 8}, type: :f32)

inputs
|> Nx.multiply(2)
|> Nx.tan