Unfortunally I don’t see Pytorch adopting Directml, with regards to ROCm I’ve been following this for over a year now and people still don’t have support for RDNA 1, maybe this will change with RDNA 2, but I doubt, it Basically Cuda, Rocm and Directml are APIs that provide fast matrix multiplication on a given platform, I like directml because on Windows at least is hardware agnostic, 'cuz it runs on top of Directx, when I heard about Rocm a few years ago I thought that this would finally be the platform to rule them all, but it runs only on Linux and in older hardware, CUDA have been very consistent that’s why I think the PyTorch team will not try to add another runtime for its framework, OpenCL would be the perfect candidate to a heterogeneous computing platform, but it didn’t work out well at the end, plaidml is something that you should look into, unfortunally PyTorch is a CUDA only framework at the moment.