Today, AMD support in PyTorch is minimal. Actually getting anything running is very difficult, and random crashes are common. This is in contrast to NVidia, which spends a lot of money to ensure a full compiler stack and compatibility with AI libraries.
Today, the AMD hardware itself is pretty capable and has a good price/performance ratio. However, actually taking advantage of that performance is difficult because of the poor quality of drivers and software.
It's going to take a big long investment, which people have been arguing about for the past 6 years, and AMD really isn't jumping up take the mantle. It's really a shame too because we need a strong competitor if we ever expect more realistic pricing for the average users/company.