Does EXLA support the Apple Silicon GPU (MPS)?

Questions & Help Questions
,
3

Ah, there’s something else you should also configure, the default defn compiler!

config :nx, :default_defn_options, compiler: EXLA should make your code quite a bit faster too