If, in the future, Norway will not deliver enough value to the USA in some way to justify that the USA pays them part of the value generated by those companies, why would the USA continue to do so? Wouldn't the USA just make their companies stop paying dividends to Norway or void Norway's shares in some way? There are probably many ways in which a country can prevent value flowing to foreign investors.
You might say "That would lead to a decline of foreign investments in the USA". True. But does the USA of the future care? If the "United States of AI" produces everything themselves, they might not care. They have a stronghold on ASML, TSMC is starting to build companies in the USA, they have Nvidia and all other GPU/TPU manufacturers, and they are investing big time into energy production.
What can keep the USA from becoming completely superior via AI and then stopping to deliver anything to the rest of the world?