Hi everyone, author here. This is a roundup from multiple projects and collaborators within the Elixir+ML ecosystem. I will be glad to answer questions where I can. :)
It runs fine, but using metal is complicated. We provide bindings for both libtorch and XLA. Libtorch has better metal support but still incomplete, with types and functions missing. Apple released a driver for XLA and it is, unfortunately, closed source and tied to a particular XLA version, so I wouldn’t particularly recommend it.