There's an important caveat, though. The software you're working with needs to support Metal/MPS, or you'll end up using the CPU instead of the GPU. For example

There's an important caveat, though. The software you're working with needs to support Metal/MPS, or you'll end up using the CPU instead of the GPU. For example, as of a few weeks ago, PyTorch still didn't support nested tensors on Metal. Probably not a deal-breaker for a lot of folks, but important to figure out before one spends thousands on hardware.