Comment by VeejayRampay

I believe pytorch works nicely with rocm, but I don't know if it's nicely to the point where it's "on par"