Hey guys, thank you for your help, it seems to have worked, I’ve already run some algorithms like MNIST and CIFAR10 here in my machine, using EXLA as a compiler, using AXON code examples, and it worked.
Unfortunately, it seems to not be processing in my GPU and still running on CPU only, but it is already an improvement.
Tomorrow I’ll investigate more and check deeper if it is actually not using GPU or if it is working as expected, and I’ll return with more info about this other issue, ok?
Please, still don’t close this thread, I would like to contribute with my implementation issues and try helping to improve EXLA/XLA documentation if you guys don’t mind.
Thx a lot, for now, @seanmor5 @josevalim , I’ve already done more progress today than in the last months trying to run EXLA with Nx.
