Stick to Nvidia if you don't want to waste your time researching for non-Nvidia solutions.

However, it's worth noting that many researchers and devs just stick to renting cloud GPUs anyway. Training usually needs something like A100 40GB or at least a T4 16GB.