Coming to the conclusion that automation and AI would target countries with higher wage costs seems to be rather obvious. The real question is what will be done to control unending Greed from turning the planet into a Welfare state.
Which will never work, UBI will never work. Why because people will never be satisfied with what they have. They will always want more. The planets resources remain limited. If its no longer a question of how hard they have to work for X; the answer to "why should I not have finer clothes, travel further faster, be warmer or be cooler, eat something nicer, etc will be that I should!"
There may be a short era of good feels, a generation that grew up working a no longer needs to and is simply satisfied with a life of comparative ease; but their grand children will demand free super sonic airline tickers, I promise you!
I am not going to pretend to know where any of this is headed. I don't think its UBI and I don't think its welfare state 2.0. I would be more worried about the collapse of states. You point out corporations are already paying armies of lobbyists to avoid taxes. You really think a group of top tier capital owner class types wont employ an army or robots that looks much more like the armies of the past and simply refuse to pay the taxes? What does for example Amazon need the government for once they can hire/build their own fully automated asset protection?