Apple MPS/Metal support?

[turian]

MPS/Metal suppport for training and inference would be amazing! There are a lot more of these devices popping up these days.

[ChinChangYang]

See this for an implementation of Metal backend. https://github.com/ChinChangYang/KataGo/releases