Generally if you want to do ML tasks you want nVidia. They put the work in early to build the tooling so now the default assumption is that you're on nVidia hardware. It is possible to do some stuff on AMD cards, but you'll be on the cutting edge for that platform re-solving problems that were already solved on the nVidia side.