It is now supported. The name of the Torch device is "mps" (for both older Intel macs with AMD GPUs and newer Macs with Apple Silicon). This is discussed in Apple docs and in this Pytorch thread. Nothing special is required, other than pip-installing the packages. (Note that the Apple docs are slightly out of date in that you no longer need to use nightly pytorch builds to have support for mps.) You can check this works with torch.rand(3, 3, device='mps'), which will fail if mps is not available.
As a tangent, for Tensorflow, you need to install the tensorflow-metal pypi dependency, discussed in Apple docs here. You can check this worked with tensorflow.config.list_physical_devices() (should include GPU:0 in the list)
LucasWiman
Users receive reputation points by contributing quality content to the forums.
Posts include post and replies published on the forums.
Post authors can mark replies as Accepted to indicate successful solutions.
Apple Developer Forums admins can mark replies as Apple Recommended to indicate an approved solution