To learn more about GPU acceleration for PyTorch and TensorFlow, please refer to Accelerate machine learning with Metal

Specifically, PyTorch is open sourced, so you can leverage this to implement custom operations in Metal. Custom ops are also supported for TensorFlow as outlined in the session.