I don't know too much about it myself, but you could checkout https://rocm.github.io/dl.html https://rocm.github.io/dl.html Edit : directly linked to the ROCm

I don't know too much about it myself, but you could checkout https://rocm.github.io/dl.html https://rocm.github.io/dl.html Edit : directly linked to the ROCm deep learning page.