In the deep learning world every major framework works on top of Cudnn which works on top of CUDA Pytorch, Tensorflow you name it.

https://github.com/pytorch/pytorch/issues/10657

That is the state of Pytorch support for AMD GPUs.