I think it's still wild that neither Tensorflow nor PyTorch work on Apple's MBP GPUs - AMD can't run ROCm on anything but Linux, and NVIDIA drivers aren't supported if you wanted to get an external GPU.
https://github.com/tensorflow/mlir/issues/60
https://github.com/tensorflow/mlir/pull/118
https://github.com/tensorflow/mlir/