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.