One of Andrew Yang's proposal was to pay for a UBI with a 10% VAT, especially on services gained from automation, AI etc. Things that give companies a productivity boost, while also avoid taxes. So this would be a way to get the money back.
For example - currently a trucker driver will pay state tax, federal tax, pay taxes on the things they eat/buy. When a self driving truck comes...all of that above won't occur anymore. And the company saves money.