function Torchx.__jit__/5 is undefined or private about nx HOT 2 CLOSED

It seems you are configuring Torchx as a compiler. Torchx is not a compiler, only a backend. If you can share the code that triggers the error, we can provide more precise feedback.

from nx.

Hi Jose,

Yes, that's right!
I was using Torchx as compiler.
I thought I needed to use it when I switched from EXLA backend to Torchx backend.
Now the loop runs without problem.

model
|> Axon.Loop.trainer(:binary_cross_entropy, :adam)
|> Axon.Loop.run(
  training_data,
  %{},
  compiler: EXLA, # Torchx,
  epochs: 10)

Thanks a lot Jose!

Best regards,

/Hiram

from nx.