[visionscaper]

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

Edit : directly linked to the ROCm deep learning page.