Hi, Would it be possible to add a context that allows for GPU acceleration on MacBook Silicon chips, similar to the [PyTorch MPS device](https://developer.apple.com/metal/pytorch/)? Cheers, Pablo