Just a after hitting fresh all-time highs, U.S. stocks are under strain as investors worry about returns on the trillions of dollars that have been put into AI.