Does EXLA support the Apple Silicon GPU (MPS)?

Questions & Help Questions
,
4

Thank you so much!

Following your advice, adding the “config :nx, :default_defn_options, compiler: EXLA” setting dramatically reduced the computation time.

It went down from 220 seconds to 13 seconds. That’s really amazing.

Thank you once again.