Any country (including the US) that tries to prevent AI from replacing some jobs will find its economy buried by another country that did not.
Conversely, a country that allows AI to replace too many jobs will find itself buried in masses of angry people with little to lose except their devalued lives.
Individuals will decide how to respond to AI, and we will see what happens. My guess, based on past adoptions of work-saving technologies, is that anyone capable of using AI to make their job easier will do so. And that may result in the elimination of other jobs.