One of the important things that NVIDIA has done in ML is make sure it is compatible with TF and Pytorch every new version they guarantee compatibility and also release their own docker containers. So you don't need to painfully install NVIDIA, CuDNN,NCCL and CUDA drivers. While AMD is releasing new hardware they still need to guarantee compatibility which NVIDIA is already years ahead. AMD barely release some initial support but not production ready