U.S. 401(k)s could feel the impact if the AI boom starts to slow