3 ms·I'm pretty sure pytorch supports it OOTB. In my experience it works even better than ROCMby mardifoufs 3y agoI'm pretty sure pytorch supports it OOTB. In my experience it works even better than ROCM