You'll want tensor cores, so I would stick to the 2000 or 3000-series Nvidia cards. Not to mention the headaches of using AMD cards with any ML framework.