How to use GPU on Apple M Mac?

[averkiev]

When I use DEVICE=auto, the Docker frontend defaults to CPU.
DEVICE=mps, results in the "PyTorch is not linked with support for mps devices" in frontend.
I've verified that PyTorch is compiled with MPS support.

Any suggestions how to offload to Macbook GPU would be welcomed.

Thanks